Spaces:
Sleeping
Sleeping
File size: 1,315 Bytes
5af269e |
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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
name="style_image_generation"
config="configs/inference_image_512_512.yaml"
ckpt="checkpoints/videocrafter_t2v_320_512/model.ckpt"
adapter_ckpt="checkpoints/stylecrafter/adapter_v1.pth"
prompt_dir="eval_data"
filename="eval_image_gen.json"
res_dir="output"
seed=123
n_samples=1
use_ddp=0
# set use_ddp=1 if you want to use multi GPU
# export CUDA_VISIBLE_DEVICES=2
if [ $use_ddp == 0 ]; then
python3 scripts/evaluation/style_inference.py \
--out_type 'image' \
--adapter_ckpt $adapter_ckpt \
--seed $seed \
--ckpt_path $ckpt \
--base $config \
--savedir $res_dir/$name \
--n_samples $n_samples \
--bs 1 --height 512 --width 512 \
--unconditional_guidance_scale 6.0 \
--ddim_steps 50 \
--ddim_eta 1.0 \
--prompt_dir $prompt_dir \
--filename $filename
fi
if [ $use_ddp == 1 ]; then
python3 -m torch.distributed.launch \
--nproc_per_node=$HOST_GPU_NUM --nnodes=$HOST_NUM --master_addr=$CHIEF_IP --master_port=23466 --node_rank=$INDEX \
scripts/evaluation/ddp_wrapper.py \
--module 'style_inference' \
--out_type 'image' \
--adapter_ckpt $adapter_ckpt \
--seed $seed \
--ckpt_path $ckpt \
--base $config \
--savedir $res_dir/$name \
--n_samples $n_samples \
--bs 1 --height 512 --width 512 \
--unconditional_guidance_scale 6.0 \
--ddim_steps 50 \
--ddim_eta 1.0 \
--prompt_dir $prompt_dir \
--filename $filename
fi |