File size: 459 Bytes
0792c6b 9f563c6 0792c6b 2e7bc51 0792c6b 2e7bc51 9f563c6 0792c6b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
export MODEL_NAME="CompVis/stable-diffusion-v1-4"
export DATA_DIR='./test_datasets/'
CUDA_VISIBLE_DEVICES=0 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 S" \
--llambda="0.8" \
--global_mapper_path="./checkpoints/global_mapper.pt" \
--local_mapper_path="./checkpoints/local_mapper.pt" \
--seed=42
|