Spaces:
Runtime error
Runtime error
File size: 1,609 Bytes
a37d690 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
#VITON-HD
##paired setting
accelerate launch inference.py --pretrained_model_name_or_path "yisol/IDM-VTON" \
--width 768 --height 1024 --num_inference_steps 30 \
--output_dir "result" --data_dir "/home/omnious/workspace/yisol/Dataset/zalando" \
--seed 42 --test_batch_size 2 --guidance_scale 2.0
##unpaired setting
accelerate launch inference.py --pretrained_model_name_or_path "yisol/IDM-VTON" \
--width 768 --height 1024 --num_inference_steps 30 \
--output_dir "result" --unpaired --data_dir "/home/omnious/workspace/yisol/Dataset/zalando" \
--seed 42 --test_batch_size 2 --guidance_scale 2.0
#DressCode
##upper_body
accelerate launch inference_dc.py --pretrained_model_name_or_path "yisol/IDM-VTON" \
--width 768 --height 1024 --num_inference_steps 30 \
--output_dir "result" --unpaired --data_dir "/home/omnious/workspace/yisol/DressCode" \
--seed 42 --test_batch_size 2 --guidance_scale 2.0 --category "upper_body"
##lower_body
accelerate launch inference_dc.py --pretrained_model_name_or_path "yisol/IDM-VTON" \
--width 768 --height 1024 --num_inference_steps 30 \
--output_dir "result" --unpaired --data_dir "/home/omnious/workspace/yisol/DressCode" \
--seed 42 --test_batch_size 2 --guidance_scale 2.0 --category "lower_body"
##dresses
accelerate launch inference_dc.py --pretrained_model_name_or_path "yisol/IDM-VTON" \
--width 768 --height 1024 --num_inference_steps 30 \
--output_dir "result" --unpaired --data_dir "/home/omnious/workspace/yisol/DressCode" \
--seed 42 --test_batch_size 2 --guidance_scale 2.0 --category "dresses"
|