Fixed randn_tensor import breaking
Browse files- 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
|
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 |
|