export CUDA_VISIBLE_DEVICES=2,3
export MODEL_NAME="/home/yangmingzhao/2024_5/sd_v1_5"
export CKPT_DIR="/home/yangmingzhao/2024_5/Fedundm/ckpt/celebs5_10_te_fix_unet_fix"
export OUTPUT_DIR="/home/yangmingzhao/2024_5/Fedundm/ckpt/celebs5_10_te_fix_unet_fix_unetdis"

python model_aggregation_unet.py --pretrained_model_name_or_path=$MODEL_NAME --ckpt_dir=${CKPT_DIR} --output_dir=${OUTPUT_DIR} --seed 77

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

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