[tests] remove autoencodertestermixin and rename new one. (#14555) · alexanderar/diffusers@4c52cb2 · GitHub
Skip to content

Commit 4c52cb2

Browse files
sayakpaulsayakpaul
authored
[tests] remove autoencodertestermixin and rename new one. (huggingface#14555)
remove autoencodertestermixin and rename new one.
1 parent 1b98ae1 commit 4c52cb2

23 files changed

Lines changed: 42 additions & 183 deletions

‎tests/models/autoencoders/test_models_asymmetric_autoencoder_kl.py‎

Lines changed: 2 additions & 2 deletions

‎tests/models/autoencoders/test_models_autoencoder_cosmos.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020

2121
from ...testing_utils import enable_full_determinism, torch_device
2222
from ..testing_utils import BaseModelTesterConfig, MemoryTesterMixin, ModelTesterMixin, TrainingTesterMixin
23-
from .testing_utils import NewAutoencoderTesterMixin
23+
from .testing_utils import AutoencoderTesterMixin
2424

2525

2626
enable_full_determinism()
@@ -92,5 +92,5 @@ class TestAutoencoderKLCosmosMemory(AutoencoderKLCosmosTesterConfig, MemoryTeste
9292
"""Memory optimization tests for AutoencoderKLCosmos."""
9393

9494

95-
class TestAutoencoderKLCosmosSlicingTiling(AutoencoderKLCosmosTesterConfig, NewAutoencoderTesterMixin):
95+
class TestAutoencoderKLCosmosSlicingTiling(AutoencoderKLCosmosTesterConfig, AutoencoderTesterMixin):
9696
"""Slicing and tiling tests for AutoencoderKLCosmos."""

‎tests/models/autoencoders/test_models_autoencoder_dc.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
SingleFileTesterMixin,
2828
TrainingTesterMixin,
2929
)
30-
from .testing_utils import NewAutoencoderTesterMixin
30+
from .testing_utils import AutoencoderTesterMixin
3131

3232

3333
enable_full_determinism()
@@ -106,7 +106,7 @@ def test_layerwise_casting_memory(self):
106106
super().test_layerwise_casting_memory()
107107

108108

109-
class TestAutoencoderDCSlicingTiling(AutoencoderDCTesterConfig, NewAutoencoderTesterMixin):
109+
class TestAutoencoderDCSlicingTiling(AutoencoderDCTesterConfig, AutoencoderTesterMixin):
110110
"""Slicing and tiling tests for AutoencoderDC."""
111111

112112

‎tests/models/autoencoders/test_models_autoencoder_hunyuan_video.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121

2222
from ...testing_utils import enable_full_determinism, torch_device
2323
from ..testing_utils import BaseModelTesterConfig, MemoryTesterMixin, ModelTesterMixin, TrainingTesterMixin
24-
from .testing_utils import NewAutoencoderTesterMixin
24+
from .testing_utils import AutoencoderTesterMixin
2525

2626

2727
enable_full_determinism()
@@ -124,7 +124,7 @@ class TestAutoencoderKLHunyuanVideoMemory(AutoencoderKLHunyuanVideoTesterConfig,
124124
"""Memory optimization tests for AutoencoderKLHunyuanVideo."""
125125

126126

127-
class TestAutoencoderKLHunyuanVideoSlicingTiling(AutoencoderKLHunyuanVideoTesterConfig, NewAutoencoderTesterMixin):
127+
class TestAutoencoderKLHunyuanVideoSlicingTiling(AutoencoderKLHunyuanVideoTesterConfig, AutoencoderTesterMixin):
128128
"""Slicing and tiling tests for AutoencoderKLHunyuanVideo."""
129129

130130
# Overwritten because the base test's block_out_channels doesn't account for the length of down_block_types.

‎tests/models/autoencoders/test_models_autoencoder_kl.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@
4242
SingleFileTesterMixin,
4343
TrainingTesterMixin,
4444
)
45-
from .testing_utils import NewAutoencoderTesterMixin
45+
from .testing_utils import AutoencoderTesterMixin
4646

4747

4848
enable_full_determinism()
@@ -203,7 +203,7 @@ class TestAutoencoderKLMemory(AutoencoderKLTesterConfig, MemoryTesterMixin):
203203
"""Memory optimization tests for AutoencoderKL."""
204204

205205

206-
class TestAutoencoderKLSlicingTiling(AutoencoderKLTesterConfig, NewAutoencoderTesterMixin):
206+
class TestAutoencoderKLSlicingTiling(AutoencoderKLTesterConfig, AutoencoderTesterMixin):
207207
"""Slicing and tiling tests for AutoencoderKL."""
208208

209209

‎tests/models/autoencoders/test_models_autoencoder_kl_cogvideox.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121

2222
from ...testing_utils import enable_full_determinism, torch_device
2323
from ..testing_utils import BaseModelTesterConfig, MemoryTesterMixin, ModelTesterMixin, TrainingTesterMixin
24-
from .testing_utils import NewAutoencoderTesterMixin
24+
from .testing_utils import AutoencoderTesterMixin
2525

2626

2727
enable_full_determinism()
@@ -107,7 +107,7 @@ class TestAutoencoderKLCogVideoXMemory(AutoencoderKLCogVideoXTesterConfig, Memor
107107
"""Memory optimization tests for AutoencoderKLCogVideoX."""
108108

109109

110-
class TestAutoencoderKLCogVideoXSlicingTiling(AutoencoderKLCogVideoXTesterConfig, NewAutoencoderTesterMixin):
110+
class TestAutoencoderKLCogVideoXSlicingTiling(AutoencoderKLCogVideoXTesterConfig, AutoencoderTesterMixin):
111111
"""Slicing and tiling tests for AutoencoderKLCogVideoX."""
112112

113113
# Overwritten because the base test's block_out_channels doesn't account for the length of down_block_types.

‎tests/models/autoencoders/test_models_autoencoder_kl_kvae.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020

2121
from ...testing_utils import enable_full_determinism, torch_device
2222
from ..testing_utils import BaseModelTesterConfig, MemoryTesterMixin, ModelTesterMixin, TrainingTesterMixin
23-
from .testing_utils import NewAutoencoderTesterMixin
23+
from .testing_utils import AutoencoderTesterMixin
2424

2525

2626
enable_full_determinism()
@@ -79,5 +79,5 @@ class TestAutoencoderKLKVAEMemory(AutoencoderKLKVAETesterConfig, MemoryTesterMix
7979
"""Memory optimization tests for AutoencoderKLKVAE."""
8080

8181

82-
class TestAutoencoderKLKVAESlicingTiling(AutoencoderKLKVAETesterConfig, NewAutoencoderTesterMixin):
82+
class TestAutoencoderKLKVAESlicingTiling(AutoencoderKLKVAETesterConfig, AutoencoderTesterMixin):
8383
"""Slicing and tiling tests for AutoencoderKLKVAE."""

‎tests/models/autoencoders/test_models_autoencoder_kl_kvae_video.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
TrainingTesterMixin,
2828
run_nondeterministic,
2929
)
30-
from .testing_utils import NewAutoencoderTesterMixin
30+
from .testing_utils import AutoencoderTesterMixin
3131

3232

3333
enable_full_determinism()
@@ -116,5 +116,5 @@ def test_layerwise_casting_training(self):
116116
run_nondeterministic(super().test_layerwise_casting_training)
117117

118118

119-
class TestAutoencoderKLKVAEVideoSlicingTiling(AutoencoderKLKVAEVideoTesterConfig, NewAutoencoderTesterMixin):
119+
class TestAutoencoderKLKVAEVideoSlicingTiling(AutoencoderKLKVAEVideoTesterConfig, AutoencoderTesterMixin):
120120
"""Slicing and tiling tests for AutoencoderKLKVAEVideo."""

‎tests/models/autoencoders/test_models_autoencoder_kl_ltx2_audio.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121

2222
from ...testing_utils import is_flaky, torch_device
2323
from ..testing_utils import BaseModelTesterConfig, MemoryTesterMixin, ModelTesterMixin, TrainingTesterMixin
24-
from .testing_utils import NewAutoencoderTesterMixin
24+
from .testing_utils import AutoencoderTesterMixin
2525

2626

2727
class AutoencoderKLLTX2AudioTesterConfig(BaseModelTesterConfig):
@@ -104,5 +104,5 @@ def test_group_offloading_with_disk(self, tmp_path, record_stream, offload_type,
104104
super().test_group_offloading_with_disk(tmp_path, record_stream, offload_type, atol=atol, rtol=rtol)
105105

106106

107-
class TestAutoencoderKLLTX2AudioSlicingTiling(AutoencoderKLLTX2AudioTesterConfig, NewAutoencoderTesterMixin):
107+
class TestAutoencoderKLLTX2AudioSlicingTiling(AutoencoderKLLTX2AudioTesterConfig, AutoencoderTesterMixin):
108108
"""Slicing and tiling tests for AutoencoderKLLTX2Audio."""

‎tests/models/autoencoders/test_models_autoencoder_kl_minimax_h3.py‎

Lines changed: 2 additions & 2 deletions

0 commit comments

Comments
 (0)