LLM/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-04-23 08:01:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tests] fix autoencoderdc tests (#13424)

Browse Source 
* fix autoencoderdc tests

* up
This commit is contained in:
Sayak Paul
2026-04-06 21:05:20 +02:00
committed by GitHub
parent 24b4c259fb
commit c39fba2ac4
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/models/autoencoders/test_models_autoencoder_dc.py
View File

@@ -28,6 +28,10 @@ enable_full_determinism()
class AutoencoderDCTesterConfig(BaseModelTesterConfig):
@property
def main_input_name(self):
return "sample"
@property
def model_class(self):
return AutoencoderDC
@@ -77,6 +81,12 @@ class AutoencoderDCTesterConfig(BaseModelTesterConfig):
class TestAutoencoderDC(AutoencoderDCTesterConfig, ModelTesterMixin):
base_precision = 1e-2
@pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16], ids=["fp16", "bf16"])
def test_from_save_pretrained_dtype_inference(self, tmp_path, dtype):
if dtype == torch.bfloat16 and IS_GITHUB_ACTIONS:
pytest.skip("Skipping bf16 test inside GitHub Actions environment")
super().test_from_save_pretrained_dtype_inference(tmp_path, dtype)
class TestAutoencoderDCTraining(AutoencoderDCTesterConfig, TrainingTesterMixin):
"""Training tests for AutoencoderDC."""