fix _optional_components in StableCascadeCombinedPipeline (#7894)

yiyixuxu · web-flow · commit 5ed3abd371d5 · 2024-05-09T06:32:55.000-10:00
* fix

* up
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

Original file line number	Diff line number	Diff line change
`@@ -153,6 +153,8 @@ def get_dummy_components(self):`
`153`	`153`	`"prior_tokenizer": prior_tokenizer,`
`154`	`154`	`"prior_prior": prior,`
`155`	`155`	`"prior_scheduler": scheduler,`
	`156`	`+ "prior_feature_extractor": None,`
	`157`	`+ "prior_image_encoder": None,`
`156`	`158`	`}`
`157`	`159`
`158`	`160`	`return components`