### train ###
bash scripts/stage1_1.sh
bash scripts/stage1_2.sh




#### infer ###
# lora
CUDA_VISIBLE_DEVICES=0 \
python inference_batch_noPrompt_fixLQ_vae_lora.py \
--model_path  \
--json_path /Pref-Restore/FFHQ_val/CelebA_HQ/captions_lq.json \
--output_dir  \
--lora_path /DiffusionNFT/logs/nft/prefRestore/multi_reward/checkpoints/checkpoint-30 \
--use_lora 

# text
CUDA_VISIBLE_DEVICES=1 \
python inference_batch_Prompt_fixLQ_vae.py \
--model_path  \
--json_path /Pref-Restore/FFHQ_val/CelebA_HQ/long_captions_lq_part.json \
--output_dir 

# w.o. text
CUDA_VISIBLE_DEVICES=1 \
python inference_batch_noPrompt_fixLQ_vae.py \
--model_path  \
--json_path /Pref-Restore/FFHQ_val/CelebA_HQ/long_captions_lq_part.json \
--output_dir