AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

stricter check.

Browse Source
This commit is contained in:
sayakpaul
2025-06-26 06:59:42 +05:30
parent e2c7d17510
commit e229ed1538
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/test_modeling_common.py
View File

@@ -1936,6 +1936,9 @@ class TorchCompileTesterMixin:
_ = model(**inputs_dict)
def test_torch_compile_repeated_blocks(self):
if self.model_class._repeated_blocks is None:
pytest.skip("Skipping test as `_repeated_blocks` is not set in the model class.")
init_dict, inputs_dict = self.prepare_init_args_and_inputs_for_common()
model = self.model_class(**init_dict).to(torch_device)