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
|