Update app.py

app.py

@@ -818,6 +818,7 @@ def main():
         pass  # run without xformers
 
     pipe = pipe.to(device)
+    pipe.unet = pipe.unet.cuda()
     prefetch_hf_cache(pipe)
     run_demo_server(pipe)