mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-12 23:48:57 +00:00
salt
80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
import os
|
|
import platform
|
|
|
|
import requests
|
|
|
|
|
|
def build_ebsynth():
|
|
if os.path.exists('deps/ebsynth/bin/ebsynth'):
|
|
print('Ebsynth has been built.')
|
|
return
|
|
|
|
os_str = platform.system()
|
|
|
|
if os_str == 'Windows':
|
|
print('Build Ebsynth Windows 64 bit.',
|
|
'If you want to build for 32 bit, please modify install.py.')
|
|
cmd = '.\\build-win64-cpu+cuda.bat'
|
|
exe_file = 'deps/ebsynth/bin/ebsynth.exe'
|
|
elif os_str == 'Linux':
|
|
cmd = 'bash build-linux-cpu+cuda.sh'
|
|
exe_file = 'deps/ebsynth/bin/ebsynth'
|
|
elif os_str == 'Darwin':
|
|
cmd = 'sh build-macos-cpu_only.sh'
|
|
exe_file = 'deps/ebsynth/bin/ebsynth.app'
|
|
else:
|
|
print('Cannot recognize OS. Ebsynth installation stopped.')
|
|
return
|
|
|
|
os.chdir('deps/ebsynth')
|
|
print(cmd)
|
|
os.system(cmd)
|
|
os.chdir('../..')
|
|
if os.path.exists(exe_file):
|
|
print('Ebsynth installed successfully.')
|
|
else:
|
|
print('Failed to install Ebsynth.')
|
|
|
|
|
|
def download(url, dir, name=None):
|
|
os.makedirs(dir, exist_ok=True)
|
|
if name is None:
|
|
name = url.split('/')[-1]
|
|
path = os.path.join(dir, name)
|
|
if not os.path.exists(path):
|
|
print(f'Install {name} ...')
|
|
open(path, 'wb').write(requests.get(url).content)
|
|
print('Install successfully.')
|
|
|
|
|
|
def download_gmflow_ckpt():
|
|
url = ('https://huggingface.co/PKUWilliamYang/Rerender/'
|
|
'resolve/main/models/gmflow_sintel-0c07dcb3.pth')
|
|
download(url, 'models')
|
|
|
|
|
|
def download_controlnet_canny():
|
|
url = ('https://huggingface.co/lllyasviel/ControlNet/'
|
|
'resolve/main/models/control_sd15_canny.pth')
|
|
download(url, 'models')
|
|
|
|
|
|
def download_controlnet_hed():
|
|
url = ('https://huggingface.co/lllyasviel/ControlNet/'
|
|
'resolve/main/models/control_sd15_hed.pth')
|
|
download(url, 'models')
|
|
|
|
|
|
def download_vae():
|
|
url = ('https://huggingface.co/stabilityai/sd-vae-ft-mse-original'
|
|
'/resolve/main/vae-ft-mse-840000-ema-pruned.ckpt')
|
|
download(url, 'models')
|
|
|
|
|
|
print("Before installing, make sure you have initialized the submodules! (git submodule init)")
|
|
build_ebsynth()
|
|
download_gmflow_ckpt()
|
|
download_controlnet_canny()
|
|
download_controlnet_hed()
|
|
download_vae()
|