lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 11:11:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update sd_forge_controlnet_example.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-27 23:14:26 -08:00
parent 33b3fe9701
commit 7ac9a06e9c
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
extensions-builtin/sd_forge_controlnet_example/scripts/sd_forge_controlnet_example.py
View File

@@ -44,16 +44,16 @@ class ControlNetExampleForge(scripts.Script):
model_dir = os.path.join(models_path, 'ControlNet') model_dir = os.path.join(models_path, 'ControlNet')
os.makedirs(model_dir, exist_ok=True) os.makedirs(model_dir, exist_ok=True)
controlnet_canny_path = load_file_from_url(
url='https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/sai_xl_canny_256lora.safetensors',
model_dir=model_dir,
file_name='sai_xl_canny_256lora.safetensors'
)
# controlnet_canny_path = load_file_from_url( # controlnet_canny_path = load_file_from_url(
# url='https://huggingface.co/lllyasviel/fav_models/resolve/main/fav/control_v11p_sd15_canny_fp16.safetensors', # url='https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/sai_xl_canny_256lora.safetensors',
# model_dir=model_dir, # model_dir=model_dir,
# file_name='control_v11p_sd15_canny_fp16.safetensors' # file_name='sai_xl_canny_256lora.safetensors'
# ) # )
controlnet_canny_path = load_file_from_url(
url='https://huggingface.co/lllyasviel/fav_models/resolve/main/fav/control_v11p_sd15_canny_fp16.safetensors',
model_dir=model_dir,
file_name='control_v11p_sd15_canny_fp16.safetensors'
)
print('The model [control_v11p_sd15_canny_fp16.safetensors] download finished.') print('The model [control_v11p_sd15_canny_fp16.safetensors] download finished.')
self.model = load_controlnet(controlnet_canny_path) self.model = load_controlnet(controlnet_canny_path)