fix dummy import for AudioLDM2 (huggingface#4741)

* fix import * style
3x0dv5 · Aug 23, 2023 · 709a642 · 709a642
1 parent 0a0fe69
commit 709a642
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/diffusers/pipelines/audioldm2/__init__.py b/src/diffusers/pipelines/audioldm2/__init__.py
@@ -11,8 +11,9 @@
         raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
     from ...utils.dummy_torch_and_transformers_objects import (
+        AudioLDM2Pipeline,
         AudioLDM2ProjectionModel,
-        AudioLDMPipeline,
+        AudioLDM2UNet2DConditionModel,
     )
 else:
     from .modeling_audioldm2 import AudioLDM2ProjectionModel, AudioLDM2UNet2DConditionModel