Skip to content

GitLab

Unverified Commit 5ed3abd3 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix `_optional_components` in `StableCascadeCombinedPipeline` (#7894) 

* fix

* up
parent 1087a510
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_combined.py
View file @ 5ed3abd3
...@@ -71,6 +71,7 @@ class StableCascadeCombinedPipeline(DiffusionPipeline): ...@@ -71,6 +71,7 @@ class StableCascadeCombinedPipeline(DiffusionPipeline):
""" """
_load_connected_pipes = True _load_connected_pipes = True
_optional_components = ["prior_feature_extractor", "prior_image_encoder"]
def __init__( def __init__(
self, self,
......
tests/pipelines/stable_cascade/test_stable_cascade_combined.py
View file @ 5ed3abd3
...@@ -153,6 +153,8 @@ class StableCascadeCombinedPipelineFastTests(PipelineTesterMixin, unittest.TestC ...@@ -153,6 +153,8 @@ class StableCascadeCombinedPipelineFastTests(PipelineTesterMixin, unittest.TestC
"prior_tokenizer": prior_tokenizer, "prior_tokenizer": prior_tokenizer,
"prior_prior": prior, "prior_prior": prior,
"prior_scheduler": scheduler, "prior_scheduler": scheduler,
"prior_feature_extractor": None,
"prior_image_encoder": None,
} }
return components return components
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment