ckpt='checkpoints/base_512_v2/model.ckpt'
config='configs/inference_t2v_512_v2.0.yaml'
PORT=$((20000 + RANDOM % 10000))

accelerate launch --multi_gpu --main_process_port $PORT scripts/main/train_t2v_lora.py \
--seed 300 \
--mode 'base' \
--ckpt_path $ckpt \
--config $config \
--height 320 --width 512 \
--unconditional_guidance_scale 12.0 \
--ddim_steps 25 \
--ddim_eta 1.0 \
--frames 12 \
--prompt_fn 'chatgpt_custom_instruments' \
--gradient_accumulation_steps 8 \
--num_train_epochs 200 \
--train_batch_size 1 \
--val_batch_size 1 \
--num_val_runs 1 \
--reward_fn 'aesthetic_hps' \
--decode_frame '-1' \
--hps_version 'v2.1' \
--lr 0.0002 \
--validation_steps 10 \
--lora_rank 16 \
--is_sample_preview True