fffiloni commited on
Commit
1225aff
1 Parent(s): 3107571

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,8 +35,8 @@ def infer(prompt, init_image, trn_steps):
35
  text_embedding_optimization_steps=500,
36
  model_fine_tuning_optimization_steps=trn_steps)
37
 
38
- # with torch.no_grad():
39
- # torch.cuda.empty_cache()
40
 
41
 
42
 
 
35
  text_embedding_optimization_steps=500,
36
  model_fine_tuning_optimization_steps=trn_steps)
37
 
38
+ with torch.no_grad():
39
+ torch.cuda.empty_cache()
40
 
41
 
42