Skip to content

Commit d1153bf

Browse files
committed
missing_modules
1 parent 0be3716 commit d1153bf

File tree

1 file changed

+3
-12
lines changed

1 file changed

+3
-12
lines changed

src/diffusers/pipelines/auto_pipeline.py

Lines changed: 3 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -529,10 +529,7 @@ def from_pipe(cls, pipeline, **kwargs):
529529
}
530530

531531
missing_modules = (
532-
set(expected_modules)
533-
- set(pipeline._optional_components)
534-
- set(text_2_image_cls._optional_components)
535-
- set(text_2_image_kwargs.keys())
532+
set(expected_modules) - set(text_2_image_cls._optional_components) - set(text_2_image_kwargs.keys())
536533
)
537534

538535
if len(missing_modules) > 0:
@@ -844,10 +841,7 @@ def from_pipe(cls, pipeline, **kwargs):
844841
}
845842

846843
missing_modules = (
847-
set(expected_modules)
848-
- set(pipeline._optional_components)
849-
- set(image_2_image_cls._optional_components)
850-
- set(image_2_image_kwargs.keys())
844+
set(expected_modules) - set(image_2_image_cls._optional_components) - set(image_2_image_kwargs.keys())
851845
)
852846

853847
if len(missing_modules) > 0:
@@ -1152,10 +1146,7 @@ def from_pipe(cls, pipeline, **kwargs):
11521146
}
11531147

11541148
missing_modules = (
1155-
set(expected_modules)
1156-
- set(pipeline._optional_components)
1157-
- set(inpainting_cls._optional_components)
1158-
- set(inpainting_kwargs.keys())
1149+
set(expected_modules) - set(inpainting_cls._optional_components) - set(inpainting_kwargs.keys())
11591150
)
11601151

11611152
if len(missing_modules) > 0:

0 commit comments

Comments
 (0)