AlanB commited on
Commit
a359d18
1 Parent(s): dfdee9a

Fixed randn_tensor import breaking

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -2
pipeline.py CHANGED
@@ -15,8 +15,8 @@ from diffusers import (
15
  UNet2DModel,
16
  )
17
  from diffusers.pipelines.unclip import UnCLIPTextProjModel
18
- from diffusers.utils import is_accelerate_available, logging, randn_tensor
19
-
20
 
21
  logger = logging.get_logger(__name__) # pylint: disable=invalid-name
22
 
 
15
  UNet2DModel,
16
  )
17
  from diffusers.pipelines.unclip import UnCLIPTextProjModel
18
+ from diffusers.utils import is_accelerate_available, logging
19
+ from diffusers.utils.torch_utils import randn_tensor
20
 
21
  logger = logging.get_logger(__name__) # pylint: disable=invalid-name
22