Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,639 Bytes
459fa69 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
#!/usr/bin/env bash
set -e
set -x
pretrained_model_name_or_path='./checkpoints/StableDiffusion'
image_encoder_path='./checkpoints/models/clip-vit-large-patch14'
controlnet_model_name_or_path='./checkpoints/models/control_v11p_sd15_lineart'
annotator_ckpts_path='./checkpoints/models/Annotators'
manga_reference_unet_path='./checkpoints/MangaNinjia/reference_unet.pth'
manga_main_model_path='./checkpoints/MangaNinjia/denoising_unet.pth'
manga_controlnet_model_path='./checkpoints/MangaNinjia/controlnet.pth'
point_net_path='./checkpoints/MangaNinjia/point_net.pth'
export CUDA_VISIBLE_DEVICES=0
input_reference_paths='./test_cases/hz0.png ./test_cases/hz1.png'
input_lineart_paths='./test_cases/hz1.png ./test_cases/hz0.png'
point_ref_paths='./test_cases/hz01_0.npy ./test_cases/hz01_1.npy'
point_lineart_paths='./test_cases/hz01_1.npy ./test_cases/hz01_0.npy'
cd ..
python infer.py \
--seed 0 \
--denoise_steps 50 \
--pretrained_model_name_or_path $pretrained_model_name_or_path --image_encoder_path $image_encoder_path \
--controlnet_model_name_or_path $controlnet_model_name_or_path --annotator_ckpts_path $annotator_ckpts_path \
--manga_reference_unet_path $manga_reference_unet_path --manga_main_model_path $manga_main_model_path \
--manga_controlnet_model_path $manga_controlnet_model_path --point_net_path $point_net_path \
--output_dir 'output' \
--guidance_scale_ref 9 \
--guidance_scale_point 15 \
--input_reference_paths $input_reference_paths \
--input_lineart_paths $input_lineart_paths \
--point_ref_paths $point_ref_paths \
--point_lineart_paths $point_lineart_paths \
|