Chain-of-Zoom / scripts /inference /inference_coz_dapeprompt.sh
alexnasa's picture
Upload 54 files
0301e15 verified
raw
history blame contribute delete
551 Bytes
#----------------- CoZ with DAPE Prompts -----------------#
# REQUIRED ENVIRONMENT: coz
INPUT_FOLDER="samples"
OUTPUT_FOLDER="inference_results/coz_dapeprompt"
CUDA_VISIBLE_DEVICES=0,1, python inference_coz.py \
-i $INPUT_FOLDER \
-o $OUTPUT_FOLDER \
--rec_type recursive \
--prompt_type dape \
--lora_path ckpt/SR_LoRA/model_20001.pkl \
--vae_path ckpt/SR_VAE/vae_encoder_20001.pt \
--pretrained_model_name_or_path 'stabilityai/stable-diffusion-3-medium-diffusers' \
--ram_ft_path ckpt/DAPE/DAPE.pth \
--ram_path ckpt/RAM/ram_swin_large_14m.pth \