Update app.py

app.py

@@ -35,8 +35,8 @@ def infer(prompt, init_image):
         text_embedding_optimization_steps=500,
         model_fine_tuning_optimization_steps=500)
     
-    with torch.no_grad():
-        torch.cuda.empty_cache()
+   # with torch.no_grad():
+   #     torch.cuda.empty_cache()