forked from nftblackmagic/IDM-VTON-training
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathinference.sh
34 lines (27 loc) · 1.61 KB
/
inference.sh
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
#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 "/workspace/MagicClothing/data/VITON-HD" \
--seed 42 --test_batch_size 1 --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"