Skip to content

Commit f7e5895

Browse files
committed
update
1 parent 54af3ca commit f7e5895

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/diffusers/pipelines/pipeline_utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1663,8 +1663,10 @@ def _get_signature_types(cls):
16631663
for k, v in inspect.signature(cls.__init__).parameters.items():
16641664
if inspect.isclass(v.annotation):
16651665
signature_types[k] = (v.annotation,)
1666-
elif get_origin(v.annotation) == Union:
1666+
elif get_origin(v.annotation) in [Union]:
16671667
signature_types[k] = get_args(v.annotation)
1668+
elif get_origin(v.annotation) in [List, Dict, list, dict]:
1669+
signature_types[k] = (v.annotation,)
16681670
else:
16691671
logger.warning(f"cannot get type annotation for Parameter {k} of {cls}.")
16701672
return signature_types

0 commit comments

Comments
 (0)