export MODEL_DIR="../modelsaved/Tiger-Corase/checkpoint-xxx"
export OUTPUT_DIR="/modelsaved/Tiger-Fine"
export DATASET="../dataset/training data/init_image"

CUDA_VISIBLE_DEVICES='0,1,2' accelerate launch --multi_gpu --num_processes=3 Tiger Model/Fine-Training.py \
     --pretrained_model_name_or_path=$MODEL_DIR \
     --output_dir=$OUTPUT_DIR \
     --dataset_name=$DATASET \
     --resolution=512 \
     --seed=2023 \
     --learning_rate=1e-5 \
     --gradient_accumulation_steps=4 \
     --checkpointing_steps=200 \
     --max_train_steps=100000000000 \
     --train_batch_size=24 \ 
     --image_column="image" \
     --caption_column_nd="text_nd" --caption_column_bg="text_bd" \
     --conditioning_nd_column="condition_FG" --conditioning_bg_column="condition_BG" \
     --validation_image "../valid_Figure.png" \
     --validation_prompt "malignant papillary solid ..." \
     --validation_steps=500