Update app.py
Browse files
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 |
-
|
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 |
|