export CUDA_VISIBLE_DEVICES=3
export MODEL_NAME="/home/yangmingzhao/2024_5/sd_v1_5"
export OUTPUT_DIR="/home/yangmingzhao/2024_5/Fedundm/ckpt/celebs50clients_20_only_unet_iimu/round0"
export IMG_DIR="/home/yangmingzhao/2024_5/Fedundm/gene/celebs50clients_20_only_unet_iimu/round0"

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