Saad0KH commited on
Commit
f2fc5e1
1 Parent(s): f46eae5

Update run/gradio_ootd.py

Browse files
Files changed (1) hide show
  1. run/gradio_ootd.py +1 -1
run/gradio_ootd.py CHANGED
@@ -62,7 +62,7 @@ def decode_image_from_base64(image_data):
62
  def process_hd(vton_img, garm_img,category, n_samples, n_steps, image_scale, seed):
63
  model_type = 'dc'
64
  with torch.no_grad():
65
- openpose_model_hd.preprocessor.body_estimation.model.to('cuda')
66
  ootd_model_hd.pipe.to('cuda')
67
  ootd_model_hd.image_encoder.to('cuda')
68
  ootd_model_hd.text_encoder.to('cuda')
 
62
  def process_hd(vton_img, garm_img,category, n_samples, n_steps, image_scale, seed):
63
  model_type = 'dc'
64
  with torch.no_grad():
65
+ openpose_model_dc.preprocessor.body_estimation.model.toopenpose_model_dc('cuda')
66
  ootd_model_hd.pipe.to('cuda')
67
  ootd_model_hd.image_encoder.to('cuda')
68
  ootd_model_hd.text_encoder.to('cuda')