fffiloni commited on
Commit
20fc321
·
verified ·
1 Parent(s): 56262ee

Update auffusion_pipeline.py

Browse files
Files changed (1) hide show
  1. 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