storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-09 00:09:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

enable auto downloading depth and openpose model

Browse Source
This commit is contained in:
Justin John
2023-11-16 20:40:15 +05:30
committed by GitHub
parent 7814c6a7e7
commit 57f88c687b
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
animation/Rerender_A_Video/install.py
+12
View File
@@ -64,6 +64,16 @@ def download_controlnet_hed():
'resolve/main/models/control_sd15_hed.pth')
download(url, 'models')
def download_controlnet_depth():
url = ('https://huggingface.co/lllyasviel/ControlNet/'
'resolve/main/models/control_sd15_depth.pth')
download(url, 'models')
def download_controlnet_openpose():
url = ('https://huggingface.co/lllyasviel/ControlNet/'
'resolve/main/models/control_sd15_openpose.pth')
download(url, 'models')
def download_vae():
url = ('https://huggingface.co/stabilityai/sd-vae-ft-mse-original'
@@ -75,4 +85,6 @@ build_ebsynth()
download_gmflow_ckpt()
download_controlnet_canny()
download_controlnet_hed()
download_controlnet_depth()
download_controlnet_openpose()
download_vae()