| export MODEL_NAME="CompVis/stable-diffusion-v1-4" | |
| export DATA_DIR='./test_datasets/' | |
| CUDA_VISIBLE_DEVICES=7 python inference_local.py \ | |
| --pretrained_model_name_or_path=$MODEL_NAME \ | |
| --test_data_dir=$DATA_DIR \ | |
| --output_dir="./outputs/local_mapping" \ | |
| --suffix="object" \ | |
| --template="a photo of a {}" \ | |
| --llambda="0.8" \ | |
| --global_mapper_path="./checkpoints/global_mapper.pt" \ | |
| --local_mapper_path="./checkpoints/local_mapper.pt" | |