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 |
|