export MODEL_DIR="" # your flux path
export OUTPUT_DIR=""  # your save path
export CONFIG="./default_config.yaml"
export LOG_PATH="$OUTPUT_DIR/log"

CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 accelerate launch --config_file $CONFIG train_kontext_lora.py \
    --pretrained_model_name_or_path $MODEL_DIR \
    --output_dir=$OUTPUT_DIR \
    --logging_dir=$LOG_PATH \
    --mixed_precision="bf16" \
    --learning_rate=1e-4 \
    --train_batch_size=1 \
    --num_train_epochs=10 \
    --validation_steps=100 \
    --checkpointing_steps=500 \
    --validation_images "./kontext_interactive_test/img_1.png" \
    --validation_prompt "Let the man hold the AK47 using both hands." \
    --num_validation_images=1