Spaces:
Sleeping
Sleeping
Update run/gradio_ootd.py
Browse files- 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 |
-
|
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')
|