Spaces:
Running
on
Zero
Running
on
Zero
Update auffusion_pipeline.py
Browse files- auffusion_pipeline.py +1 -1
auffusion_pipeline.py
CHANGED
|
@@ -26,12 +26,12 @@ from diffusers.image_processor import VaeImageProcessor
|
|
| 26 |
from diffusers.loaders import FromSingleFileMixin, LoraLoaderMixin, TextualInversionLoaderMixin
|
| 27 |
from diffusers.models import AutoencoderKL, UNet2DConditionModel
|
| 28 |
from diffusers.schedulers import KarrasDiffusionSchedulers
|
|
|
|
| 29 |
from diffusers.utils import (
|
| 30 |
deprecate,
|
| 31 |
is_accelerate_available,
|
| 32 |
is_accelerate_version,
|
| 33 |
logging,
|
| 34 |
-
randn_tensor,
|
| 35 |
replace_example_docstring,
|
| 36 |
)
|
| 37 |
from diffusers.pipelines.pipeline_utils import DiffusionPipeline
|
|
|
|
| 26 |
from diffusers.loaders import FromSingleFileMixin, LoraLoaderMixin, TextualInversionLoaderMixin
|
| 27 |
from diffusers.models import AutoencoderKL, UNet2DConditionModel
|
| 28 |
from diffusers.schedulers import KarrasDiffusionSchedulers
|
| 29 |
+
from diffusers.utils.torch_utils import randn_tensor
|
| 30 |
from diffusers.utils import (
|
| 31 |
deprecate,
|
| 32 |
is_accelerate_available,
|
| 33 |
is_accelerate_version,
|
| 34 |
logging,
|
|
|
|
| 35 |
replace_example_docstring,
|
| 36 |
)
|
| 37 |
from diffusers.pipelines.pipeline_utils import DiffusionPipeline
|