fix

huggingface · Dec 23, 2024 · a5e6c13 · a5e6c13
1 parent 178c22d
commit a5e6c13
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/tests/lora/test_lora_layers_ltx_video.py b/tests/lora/test_lora_layers_ltx_video.py
@@ -52,10 +52,19 @@ class LTXVideoLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
     }
     transformer_cls = LTXVideoTransformer3DModel
     vae_kwargs = {
+        "in_channels": 3,
+        "out_channels": 3,
         "latent_channels": 8,
         "block_out_channels": (8, 8, 8, 8),
-        "spatio_temporal_scaling": (True, True, False, False),
+        "decoder_block_out_channels": (8, 8, 8, 8),
         "layers_per_block": (1, 1, 1, 1, 1),
+        "decoder_layers_per_block": (1, 1, 1, 1, 1),
+        "spatio_temporal_scaling": (True, True, False, False),
+        "decoder_spatio_temporal_scaling": (True, True, False, False),
+        "decoder_inject_noise": (False, False, False, False, False),
+        "upsample_residual": (False, False, False, False),
+        "upsample_factor": (1, 1, 1, 1),
+        "timestep_conditioning": False,
         "patch_size": 1,
         "patch_size_t": 1,
         "encoder_causal": True,