Skip to content

Commit f243282

Browse files
[Dummy imports] Add missing if else statements for SD] (huggingface#2381)
* [Dummy imports] Add missing if else statements for SD] * Apply suggestions from code review Co-authored-by: Pedro Cuenca <[email protected]> --------- Co-authored-by: Pedro Cuenca <[email protected]>
1 parent ca980fd commit f243282

File tree

1 file changed

+18
-3
lines changed

1 file changed

+18
-3
lines changed

src/diffusers/pipelines/stable_diffusion/__init__.py

Lines changed: 18 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,12 @@ class StableDiffusionPipelineOutput(BaseOutput):
3636
nsfw_content_detected: Optional[List[bool]]
3737

3838

39-
if is_transformers_available() and is_torch_available():
39+
try:
40+
if not (is_transformers_available() and is_torch_available()):
41+
raise OptionalDependencyNotAvailable()
42+
except OptionalDependencyNotAvailable:
43+
from ...utils.dummy_torch_and_transformers_objects import * # noqa F403
44+
else:
4045
from .pipeline_cycle_diffusion import CycleDiffusionPipeline
4146
from .pipeline_stable_diffusion import StableDiffusionPipeline
4247
from .pipeline_stable_diffusion_img2img import StableDiffusionImg2ImgPipeline
@@ -74,14 +79,24 @@ class StableDiffusionPipelineOutput(BaseOutput):
7479

7580

7681
try:
77-
if not (is_torch_available() and is_transformers_available() and is_k_diffusion_version(">=", "0.0.12")):
82+
if not (
83+
is_torch_available()
84+
and is_transformers_available()
85+
and is_k_diffusion_available()
86+
and is_k_diffusion_version(">=", "0.0.12")
87+
):
7888
raise OptionalDependencyNotAvailable()
7989
except OptionalDependencyNotAvailable:
8090
from ...utils.dummy_torch_and_transformers_and_k_diffusion_objects import * # noqa F403
8191
else:
8292
from .pipeline_stable_diffusion_k_diffusion import StableDiffusionKDiffusionPipeline
8393

84-
if is_transformers_available() and is_onnx_available():
94+
try:
95+
if not (is_transformers_available() and is_onnx_available()):
96+
raise OptionalDependencyNotAvailable()
97+
except OptionalDependencyNotAvailable:
98+
from ...utils.dummy_onnx_objects import * # noqa F403
99+
else:
85100
from .pipeline_onnx_stable_diffusion import OnnxStableDiffusionPipeline, StableDiffusionOnnxPipeline
86101
from .pipeline_onnx_stable_diffusion_img2img import OnnxStableDiffusionImg2ImgPipeline
87102
from .pipeline_onnx_stable_diffusion_inpaint import OnnxStableDiffusionInpaintPipeline

0 commit comments

Comments
 (0)