diff --git a/tests/lora/test_lora_layers_ltx_video.py b/tests/lora/test_lora_layers_ltx_video.py index 1ed426f6e8dd..0eccaa73ad42 100644 --- 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,