ashawkey commited on
Commit
66cb594
β€’
1 Parent(s): 218cd4f
Files changed (1) hide show
  1. mvdream/pipeline_mvdream.py +1 -1
mvdream/pipeline_mvdream.py CHANGED
@@ -499,6 +499,7 @@ class MVDreamPipeline(DiffusionPipeline):
499
  # Prepare extra step kwargs.
500
  extra_step_kwargs = self.prepare_extra_step_kwargs(generator, eta)
501
 
 
502
  context = torch.cat([prompt_embeds_neg] * actual_num_frames + [prompt_embeds_pos] * actual_num_frames)
503
  torch.cat([camera] * multiplier)
504
 
@@ -511,7 +512,6 @@ class MVDreamPipeline(DiffusionPipeline):
511
  with self.progress_bar(total=num_inference_steps) as progress_bar:
512
  for i, t in enumerate(timesteps):
513
  # expand the latents if we are doing classifier free guidance
514
- multiplier = 2 if do_classifier_free_guidance else 1
515
  latent_model_input = torch.cat([latents] * multiplier)
516
  latent_model_input = self.scheduler.scale_model_input(latent_model_input, t)
517
 
 
499
  # Prepare extra step kwargs.
500
  extra_step_kwargs = self.prepare_extra_step_kwargs(generator, eta)
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
 
 
512
  with self.progress_bar(total=num_inference_steps) as progress_bar:
513
  for i, t in enumerate(timesteps):
514
  # expand the latents if we are doing classifier free guidance
 
515
  latent_model_input = torch.cat([latents] * multiplier)
516
  latent_model_input = self.scheduler.scale_model_input(latent_model_input, t)
517