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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ 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
 
 
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