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

up

Browse Source
This commit is contained in:
sayakpaul
2025-11-28 19:13:08 +05:30
parent 1b0888c24e
commit 7c47ae0899
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/test_modeling_common.py
View File

@@ -2193,6 +2193,8 @@ class TorchCompileTesterMixin:
recompile_limit = 1
if self.model_class.__name__ == "UNet2DConditionModel":
recompile_limit = 2
elif self.model_class.__name__ == "ZImageTransformer2DModel":
recompile_limit = 3
with (
torch._inductor.utils.fresh_inductor_cache(),
@@ -2294,7 +2296,6 @@ class LoraHotSwappingForModelTesterMixin:
backend_empty_cache(torch_device)
def get_lora_config(self, lora_rank, lora_alpha, target_modules):
# from diffusers test_models_unet_2d_condition.py
from peft import LoraConfig
lora_config = LoraConfig(