Skip to content

GitLab

Unverified Commit 709a6428 authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

fix dummy import for AudioLDM2 (#4741) 

* fix import

* style
parent 0a0fe69a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
src/diffusers/pipelines/audioldm2/__init__.py
View file @ 709a6428
...@@ -11,8 +11,9 @@ try: ...@@ -11,8 +11,9 @@ try:
raise OptionalDependencyNotAvailable() raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable: except OptionalDependencyNotAvailable:
from ...utils.dummy_torch_and_transformers_objects import ( from ...utils.dummy_torch_and_transformers_objects import (
AudioLDM2Pipeline,
AudioLDM2ProjectionModel, AudioLDM2ProjectionModel,
AudioLDMPipeline, AudioLDM2UNet2DConditionModel,
) )
else: else:
from .modeling_audioldm2 import AudioLDM2ProjectionModel, AudioLDM2UNet2DConditionModel from .modeling_audioldm2 import AudioLDM2ProjectionModel, AudioLDM2UNet2DConditionModel
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment