export CUDA_VISIBLE_DEVICES=0,1
export MODEL_NAME="/home/yangmingzhao/2024_5/sd_v1_5"
export OUTPUT_DIR="/home/yangmingzhao/2024_5/Fedundm/ckpt/celebsclients_15_linear_forget_iimu_randompush"
export IMG_DIR="/home/yangmingzhao/2024_5/Fedundm/gene/celebsclients_15_linear_forget_iimu_randompush"

#accelerate launch --mixed_precision="fp16"  train_text_encoder_lora.py \
#  --pretrained_model_name_or_path=$MODEL_NAME \
#  --output_dir=${OUTPUT_DIR} \
#  --seed=77 \
#  --round=$i
mkdir ${OUTPUT_DIR}
cp *.py ${OUTPUT_DIR}
cp lora.sh ${OUTPUT_DIR}

accelerate launch --mixed_precision="fp16"  --main_process_port=29501 train_unet_lora_celebs.py \
  --pretrained_model_name_or_path=$MODEL_NAME \
  --output_dir=${OUTPUT_DIR} \
  --seed=12345 \
  --round=0
  
#python model_aggregation_avg_celebs.py --output_dir=${OUTPUT_DIR} --round=0

#python gene_image.py --ckpt_dir=${OUTPUT_DIR} --output_dir=${IMG_DIR} --seed=52+$i

#python agg_gene_image.py --ckpt_dir=${OUTPUT_DIR} --output_dir=${IMG_DIR} --seed=52+$i