Spaces:
Running
on
Zero
Running
on
Zero
fix
Browse files
mvdream/pipeline_mvdream.py
CHANGED
@@ -501,7 +501,7 @@ class MVDreamPipeline(DiffusionPipeline):
|
|
501 |
|
502 |
multiplier = 2 if do_classifier_free_guidance else 1
|
503 |
context = torch.cat([prompt_embeds_neg] * actual_num_frames + [prompt_embeds_pos] * actual_num_frames)
|
504 |
-
torch.cat([camera] * multiplier)
|
505 |
|
506 |
if image is not None:
|
507 |
ip = torch.cat([image_embeds_neg] * actual_num_frames + [image_embeds_pos] * actual_num_frames)
|
|
|
501 |
|
502 |
multiplier = 2 if do_classifier_free_guidance else 1
|
503 |
context = torch.cat([prompt_embeds_neg] * actual_num_frames + [prompt_embeds_pos] * actual_num_frames)
|
504 |
+
camera = torch.cat([camera] * multiplier)
|
505 |
|
506 |
if image is not None:
|
507 |
ip = torch.cat([image_embeds_neg] * actual_num_frames + [image_embeds_pos] * actual_num_frames)
|