fix dummy import for AudioLDM2 (#4741)

* fix import * style

fix dummy import for AudioLDM2 (#4741)
* fix import * style
709a6428 · Suraj Patil · GitHub · 0a0fe69a · 709a6428
Unverified Commit 709a6428 authored Aug 23, 2023 by Suraj Patil Committed by GitHub Aug 23, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/diffusers/pipelines/audioldm2/__init__.py src/diffusers/pipelines/audioldm2/__init__.py +2 -1

No files found.
--- a/src/diffusers/pipelines/audioldm2/__init__.py
+++ b/src/diffusers/pipelines/audioldm2/__init__.py
@@ -11,8 +11,9 @@ try:
        raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
    from ...utils.dummy_torch_and_transformers_objects import (
+        AudioLDM2Pipeline,
        AudioLDM2ProjectionModel,
-        AudioLDMPipeline,
+        AudioLDM2UNet2DConditionModel,
    )
 else:
    from .modeling_audioldm2 import AudioLDM2ProjectionModel, AudioLDM2UNet2DConditionModel