From 5ee4e19c58a751036dec203ad1fe08d68894ff63 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Tue, 13 Jan 2026 08:57:21 +0530 Subject: [PATCH] handle modern types. --- src/diffusers/pipelines/pipeline_utils.py | 4 ++++ tests/lora/test_lora_layers_wan.py | 13 ++----------- 2 files changed, 6 insertions(+), 11 deletions(-) diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py index d9ec1466c6..35e1c35745 100644 --- a/src/diffusers/pipelines/pipeline_utils.py +++ b/src/diffusers/pipelines/pipeline_utils.py @@ -19,6 +19,7 @@ import inspect import os import re import sys +import types from dataclasses import dataclass from pathlib import Path from typing import Any, Callable, Dict, List, Union, get_args, get_origin @@ -1823,6 +1824,9 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin): signature_types[k] = (v.annotation,) elif get_origin(v.annotation) == Union: signature_types[k] = get_args(v.annotation) + elif isinstance(v.annotation, types.UnionType): + # Handle PEP 604 union syntax (X | Y) introduced in Python 3.10+ + signature_types[k] = get_args(v.annotation) elif get_origin(v.annotation) in [List, Dict, list, dict]: signature_types[k] = (v.annotation,) else: diff --git a/tests/lora/test_lora_layers_wan.py b/tests/lora/test_lora_layers_wan.py index 5734509b41..dde486b27f 100644 --- a/tests/lora/test_lora_layers_wan.py +++ b/tests/lora/test_lora_layers_wan.py @@ -18,18 +18,9 @@ import unittest import torch from transformers import AutoTokenizer, T5EncoderModel -from diffusers import ( - AutoencoderKLWan, - FlowMatchEulerDiscreteScheduler, - WanPipeline, - WanTransformer3DModel, -) +from diffusers import AutoencoderKLWan, FlowMatchEulerDiscreteScheduler, WanPipeline, WanTransformer3DModel -from ..testing_utils import ( - floats_tensor, - require_peft_backend, - skip_mps, -) +from ..testing_utils import floats_tensor, require_peft_backend, skip_mps sys.path.append(".")