Type annotation fix (#11597)

* update * update

Type annotation fix (#11597)
* update * update
c36f8487 · Dhruv Nair · GitHub · 54af3ca7 · c36f8487
Unverified Commit c36f8487 authored May 21, 2025 by Dhruv Nair Committed by GitHub May 21, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/diffusers/pipelines/pipeline_utils.py src/diffusers/pipelines/pipeline_utils.py +2 -0

No files found.
--- a/src/diffusers/pipelines/pipeline_utils.py
+++ b/src/diffusers/pipelines/pipeline_utils.py
@@ -1665,6 +1665,8 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin):
                signature_types[k] = (v.annotation,)
            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,)
            else:
                logger.warning(f"cannot get type annotation for Parameter {k} of {cls}.")
        return signature_types