Update pipeline_imagedream.py
Browse files- pipeline_imagedream.py +1 -1
pipeline_imagedream.py
CHANGED
@@ -193,7 +193,7 @@ class ImageDreamPipeline(StableDiffusionPipeline):
|
|
193 |
if cross_attention_kwargs is None:
|
194 |
num_views = self.num_views
|
195 |
else:
|
196 |
-
cross_attention_kwargs.pop("num_views", self.num_views)
|
197 |
|
198 |
# 0. Default height and width to unet
|
199 |
height = height or self.unet.config.sample_size * self.vae_scale_factor
|
|
|
193 |
if cross_attention_kwargs is None:
|
194 |
num_views = self.num_views
|
195 |
else:
|
196 |
+
num_views = cross_attention_kwargs.pop("num_views", self.num_views)
|
197 |
|
198 |
# 0. Default height and width to unet
|
199 |
height = height or self.unet.config.sample_size * self.vae_scale_factor
|