AP123 commited on
Commit
0850a5c
1 Parent(s): b295b08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ def call(
63
  else:
64
  latent_model_input2 = torch.cat([latents.flip(2)] * 2) if do_classifier_free_guidance else latents
65
  latent_model_input2 = pipe.scheduler.scale_model_input(latent_model_input2, t)
66
- noise_pred2 = pipe.unet(latent_model_input2
67
 
68
  def simple_call(prompt1, prompt2, guidance_scale1, guidance_scale2, negative_prompt1, negative_prompt2):
69
  generator = [torch.Generator(device="cuda").manual_seed(5)]
 
63
  else:
64
  latent_model_input2 = torch.cat([latents.flip(2)] * 2) if do_classifier_free_guidance else latents
65
  latent_model_input2 = pipe.scheduler.scale_model_input(latent_model_input2, t)
66
+ noise_pred2 = pipe.unet(latent_model_input2, t)
67
 
68
  def simple_call(prompt1, prompt2, guidance_scale1, guidance_scale2, negative_prompt1, negative_prompt2):
69
  generator = [torch.Generator(device="cuda").manual_seed(5)]