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

up

Browse Source
This commit is contained in:
sayakpaul
2025-10-27 19:09:40 +05:30
parent 4490e4cc44
commit bcada5bfaf
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/diffusers/utils/typing_utils.py
View File

@@ -18,6 +18,12 @@ Typing utilities: Utilities related to type checking and validation
from typing import Any, Set, Type, Union, get_args, get_origin
try:
from types import UnionType as _UnionType
except ImportError: # Python < 3.10
_UnionType = None
def _is_valid_type(obj: Any, class_or_tuple: Type | tuple[Type, ...]) -> bool:
"""
Checks if an object is an instance of any of the provided types. For collections, it checks if every element is of
@@ -29,7 +35,12 @@ def _is_valid_type(obj: Any, class_or_tuple: Type | tuple[Type, ...]) -> bool:
# Unpack unions
unpacked_class_or_tuple = []
for t in class_or_tuple:
if get_origin(t) is Union:
origin = get_origin(t)
is_union = origin is Union or (_UnionType is not None and origin is _UnionType)
# For PEP 604 unions (e.g. int | float), origin can be None but the object itself is a UnionType
if not is_union and _UnionType is not None and isinstance(t, _UnionType):
is_union = True
if is_union:
unpacked_class_or_tuple.extend(get_args(t))
else:
unpacked_class_or_tuple.append(t)