export CUDA_VISIBLE_DEVICES=6
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/round0"
export IMG_DIR="/home/yangmingzhao/2024_5/Fedundm/gene/celebsclients_15_linear_forget_iimu_randompush_onlylast/round0"

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