@@ -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
7681try :
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 ()
7989except OptionalDependencyNotAvailable :
8090 from ...utils .dummy_torch_and_transformers_and_k_diffusion_objects import * # noqa F403
8191else :
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