Update app.py

app.py

@@ -63,11 +63,11 @@ def run(ref_path, ref_style, ref_prompt, prompt1, prompt2, prompt3):
 
     zT, inversion_callback = inversion.make_inversion_callback(zts, offset=5)
 
-    g_cpu = torch.Generator(device='cpu')
+    g_cpu = torch.Generator(device='cuda')
     g_cpu.manual_seed(10)
 
-    latents = torch.randn(len(prompts), 4, 128, 128, device='cpu', generator=g_cpu,
-                      dtype=pipeline.unet.dtype,).to('cuda:0')
+    latents = torch.randn(len(prompts), 4, 128, 128, device='cuda', generator=g_cpu,
+                      dtype=pipeline.unet.dtype,).to('cuda')
     latents[0] = zT
 
     images_a = pipeline(prompts, latents=latents,