Update app.py

app.py

@@ -61,7 +61,9 @@ def infer(prompt,negative_prompt,seed,resolution, steps):
     
     # generator = torch.Generator("cuda").manual_seed(555)
     t=time.time()
-    pipe.to("cuda")
+    pipe.unet.unet.to("cuda")
+    pipe.unet.ella.to("cuda")
+    
     if seed=="-1":
         generator=None
     else: