generate_test_sample.sh

#!/bin/bash

model_root="/home/vtryon/catvton_v0/checkpoints/32000_openpose_bak"
data_record_path="/home/vtryon/catvton_v0/datasets/test/test_data.jsonl"
vae_subfolder="vae"
output_dir="../test_outputs"
height=512
width=384
extra_condition_key="openpose"
batch_size=16
weight_dtype="fp16"
num_inference_steps=50
guidance_scale=0
data_nums=8

accelerate launch --main_process_port=12321 generate_test_sample.py \
--model_root=${model_root} \
--data_record_path=${data_record_path} \
--vae_subfolder=${vae_subfolder} \
--output_dir=${output_dir} \
--height=${height} \
--width=${width} \
--extra_condition_key=${extra_condition_key} \
--batch_size=${batch_size} \
--weight_dtype=${weight_dtype} \
--num_inference_steps=${num_inference_steps} \
--guidance_scale=${guidance_scale} \
--repaint