merge main

src/diffusers/__init__.py

@@ -6,6 +6,7 @@
     OptionalDependencyNotAvailable,
     is_flax_available,
     is_inflect_available,
+    is_k_diffusion_available,
     is_librosa_available,
     is_onnx_available,
     is_scipy_available,
@@ -100,6 +101,13 @@
         VQDiffusionPipeline,
     )
 
+try:
+    if not (is_torch_available() and is_transformers_available() and is_k_diffusion_available()):
+        raise OptionalDependencyNotAvailable()
+except OptionalDependencyNotAvailable:
+    from .utils.dummy_torch_and_transformers_and_onnx_objects import *  # noqa F403
+else:
+    from .pipelines import StableDiffusionKDiffusionPipeline
 
 try:
     if not (is_torch_available() and is_transformers_available() and is_onnx_available()):

src/diffusers/pipelines/__init__.py

@@ -74,6 +74,15 @@
         StableDiffusionOnnxPipeline,
     )
 
+try:
+    if not (is_torch_available() and is_transformers_available() and is_k_diffusion_available()):
+        raise OptionalDependencyNotAvailable()
+except OptionalDependencyNotAvailable:
+    from ..utils.dummy_torch_and_transformers_and_k_diffusion_objects import *  # noqa F403
+else:
+    from .stable_diffusion import StableDiffusionKDiffusionPipeline
+
+
 try:
     if not (is_flax_available() and is_transformers_available()):
         raise OptionalDependencyNotAvailable()