1 files changed, 2 insertions, 2 deletions

diff --git a/pipelines/stable_diffusion/vlpn_stable_diffusion.py b/pipelines/stable_diffusion/vlpn_stable_diffusion.py
index f02dd72..5f4fc38 100644
--- a/pipelines/stable_diffusion/vlpn_stable_diffusion.py
+++ b/pipelines/stable_diffusion/vlpn_stable_diffusion.py
@@ -22,7 +22,7 @@ from diffusers import (
     PNDMScheduler,
 )
 from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion import StableDiffusionPipelineOutput
-from diffusers.utils import logging, randn_tensor
+from diffusers.utils import logging
 from transformers import CLIPTextModel, CLIPTokenizer
 
 from models.clip.util import unify_input_ids, get_extended_embeddings
@@ -308,7 +308,7 @@ class VlpnStableDiffusion(DiffusionPipeline):
 
     def prepare_image(self, batch_size, width, height, dtype, device, generator=None):
         noise = perlin_noise(
-            batch_size, width, height, res=1, octaves=4, generator=generator, dtype=dtype, device=device
+            batch_size, 1, width, height, res=1, octaves=4, generator=generator, dtype=dtype, device=device
         ).expand(batch_size, 3, width, height)
         return (1.4 * noise).clamp(-1, 1)