From f7e5895044077585b45c6874de874038c57968da Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Wed, 21 May 2025 16:21:06 +0200 Subject: [PATCH 1/2] update --- src/diffusers/pipelines/pipeline_utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py index 3d92034aef15..17cf06b58a50 100644 --- a/src/diffusers/pipelines/pipeline_utils.py +++ b/src/diffusers/pipelines/pipeline_utils.py @@ -1663,8 +1663,10 @@ def _get_signature_types(cls): for k, v in inspect.signature(cls.__init__).parameters.items(): if inspect.isclass(v.annotation): signature_types[k] = (v.annotation,) - elif get_origin(v.annotation) == Union: + elif get_origin(v.annotation) in [Union]: signature_types[k] = get_args(v.annotation) + elif get_origin(v.annotation) in [List, Dict, list, dict]: + signature_types[k] = (v.annotation,) else: logger.warning(f"cannot get type annotation for Parameter {k} of {cls}.") return signature_types From b1c4d8ce850fc6dbb1c9495569baf64d2dc87333 Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Wed, 21 May 2025 17:34:38 +0200 Subject: [PATCH 2/2] update --- src/diffusers/pipelines/pipeline_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py index 17cf06b58a50..8184573b02ef 100644 --- a/src/diffusers/pipelines/pipeline_utils.py +++ b/src/diffusers/pipelines/pipeline_utils.py @@ -1663,7 +1663,7 @@ def _get_signature_types(cls): for k, v in inspect.signature(cls.__init__).parameters.items(): if inspect.isclass(v.annotation): signature_types[k] = (v.annotation,) - elif get_origin(v.annotation) in [Union]: + elif get_origin(v.annotation) == Union: signature_types[k] = get_args(v.annotation) elif get_origin(v.annotation) in [List, Dict, list, dict]: signature_types[k] = (v.annotation,)