fix _optional_components in StableCascadeCombinedPipeline (#7894)

* fix * up
huggingface · May 9, 2024 · 5ed3abd · 5ed3abd
1 parent 1087a51
commit 5ed3abd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_combined.py b/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_combined.py
@@ -71,6 +71,7 @@ class StableCascadeCombinedPipeline(DiffusionPipeline):
     """
 
     _load_connected_pipes = True
+    _optional_components = ["prior_feature_extractor", "prior_image_encoder"]
 
     def __init__(
         self,

diff --git a/tests/pipelines/stable_cascade/test_stable_cascade_combined.py b/tests/pipelines/stable_cascade/test_stable_cascade_combined.py
@@ -153,6 +153,8 @@ def get_dummy_components(self):
             "prior_tokenizer": prior_tokenizer,
             "prior_prior": prior,
             "prior_scheduler": scheduler,
+            "prior_feature_extractor": None,
+            "prior_image_encoder": None,
         }
 
         return components