rafiaashraf commited on
Commit
4dd2933
Β·
verified Β·
1 Parent(s): 03a33c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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)