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-29 08:14:17 +05:30
parent 66b69229f8
commit 2c367f84f4
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

@@ -47,6 +47,7 @@ from diffusers.models.attention_processor import (
XFormersAttnProcessor,
)
from diffusers.models.auto_model import AutoModel
from diffusers.models.modeling_outputs import BaseOutput
from diffusers.training_utils import EMAModel
from diffusers.utils import (
SAFE_WEIGHTS_INDEX_NAME,
@@ -109,7 +110,7 @@ def check_if_lora_correctly_set(model) -> bool:
def normalize_output(out):
out0 = out[0] if isinstance(out, tuple) else out
out0 = out[0] if isinstance(out, (BaseOutput, tuple)) else out
return torch.stack(out0) if isinstance(out0, list) else out0