update

huggingface · Dec 23, 2024 · 178c22d · 178c22d
1 parent 4325449
commit 178c22d
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/tests/models/autoencoders/test_models_autoencoder_ltx_video.py b/tests/models/autoencoders/test_models_autoencoder_ltx_video.py
@@ -82,11 +82,11 @@ def prepare_init_args_and_inputs_for_common(self):
 
     def test_gradient_checkpointing_is_applied(self):
         expected_set = {
-            "LTXEncoder3d",
-            "LTXDecoder3d",
-            "LTXDownBlock3D",
-            "LTXMidBlock3d",
-            "LTXUpBlock3d",
+            "LTXVideoEncoder3d",
+            "LTXVideoDecoder3d",
+            "LTXVideoDownBlock3D",
+            "LTXVideoMidBlock3d",
+            "LTXVideoUpBlock3d",
         }
         super().test_gradient_checkpointing_is_applied(expected_set=expected_set)
 
@@ -152,11 +152,11 @@ def prepare_init_args_and_inputs_for_common(self):
 
     def test_gradient_checkpointing_is_applied(self):
         expected_set = {
-            "LTXEncoder3d",
-            "LTXDecoder3d",
-            "LTXDownBlock3D",
-            "LTXMidBlock3d",
-            "LTXUpBlock3d",
+            "LTXVideoEncoder3d",
+            "LTXVideoDecoder3d",
+            "LTXVideoDownBlock3D",
+            "LTXVideoMidBlock3d",
+            "LTXVideoUpBlock3d",
         }
         super().test_gradient_checkpointing_is_applied(expected_set=expected_set)