kiigii
/

imagedream-ipmv-diffusers

Model card Files Files and versions

kiigii commited on Jul 3, 2024

Commit

184243f

·

verified ·

1 Parent(s): 3ae12b5

Update pipeline_imagedream.py

Files changed (1) hide show

pipeline_imagedream.py +1 -1

pipeline_imagedream.py CHANGED Viewed

@@ -193,7 +193,7 @@ class ImageDreamPipeline(StableDiffusionPipeline):
         if cross_attention_kwargs is None:
             num_views = self.num_views
         else:
-            cross_attention_kwargs.pop("num_views", self.num_views)
         # 0. Default height and width to unet
         height = height or self.unet.config.sample_size * self.vae_scale_factor

         if cross_attention_kwargs is None:
             num_views = self.num_views
         else:
+            num_views = cross_attention_kwargs.pop("num_views", self.num_views)
         # 0. Default height and width to unet
         height = height or self.unet.config.sample_size * self.vae_scale_factor