minor changes

requirements.txt

@@ -1,7 +1,7 @@
 accelerate
 datasets
 diffusers==0.24.0
-numpy==1.26.1
+numpy==1.26.4
 packaging==23.2
 pandas_stubs==1.2.0.57
 Pillow==10.1.0

src/pipelines/pipeline_kandinsky_prior.py

@@ -18,11 +18,11 @@ from diffusers.utils import (
     is_accelerate_available,
     is_accelerate_version,
     logging,
-    randn_tensor,
+    # randn_tensor,
     replace_example_docstring,
 )
 from diffusers.pipelines.pipeline_utils import DiffusionPipeline
-
+from diffusers.utils.torch_utils import randn_tensor
 
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name
 

src/pipelines/pipeline_unclip.py

@@ -12,7 +12,8 @@ from transformers.models.clip.modeling_clip import CLIPTextModelOutput
 
 from diffusers.models import UNet2DConditionModel, UNet2DModel
 from diffusers.schedulers import UnCLIPScheduler
-from diffusers.utils import logging, randn_tensor
+from diffusers.utils import logging #, randn_tensor
+from diffusers.utils.torch_utils import randn_tensor
 from diffusers.pipelines.pipeline_utils import DiffusionPipeline, ImagePipelineOutput
 from diffusers.pipelines.unclip.text_proj import UnCLIPTextProjModel