Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -126,6 +126,8 @@ def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_ste
|
|
| 126 |
print("start_tryon started correctly")
|
| 127 |
device = "cuda"
|
| 128 |
print("CUDA version:", torch.version.cuda)
|
|
|
|
|
|
|
| 129 |
|
| 130 |
openpose_model.preprocessor.body_estimation.model.to(device)
|
| 131 |
pipe.to(device)
|
|
|
|
| 126 |
print("start_tryon started correctly")
|
| 127 |
device = "cuda"
|
| 128 |
print("CUDA version:", torch.version.cuda)
|
| 129 |
+
torch.cuda.empty_cache()
|
| 130 |
+
torch.cuda.ipc_collect()
|
| 131 |
|
| 132 |
openpose_model.preprocessor.body_estimation.model.to(device)
|
| 133 |
pipe.to(device)
|