From 178c22dc51dfa164e0f334a2c2beb87898eaca60 Mon Sep 17 00:00:00 2001 From: Aryan Date: Mon, 23 Dec 2024 11:20:37 +0100 Subject: [PATCH] update --- .../test_models_autoencoder_ltx_video.py | 20 +++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/tests/models/autoencoders/test_models_autoencoder_ltx_video.py b/tests/models/autoencoders/test_models_autoencoder_ltx_video.py index 01a8c0e77806..37f9837c8245 100644 --- 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)