Update app.py

app.py

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