Spaces:
Runtime error
Runtime error
| import os | |
| import platform | |
| import requests | |
| def build_ebsynth(): | |
| if os.path.exists('src/ebsynth/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 = 'src/ebsynth/deps/ebsynth/bin/ebsynth.exe' | |
| elif os_str == 'Linux': | |
| cmd = 'bash build-linux-cpu+cuda.sh' | |
| exe_file = 'src/ebsynth/deps/ebsynth/bin/ebsynth' | |
| elif os_str == 'Darwin': | |
| cmd = 'sh build-macos-cpu_only.sh' | |
| exe_file = 'src/ebsynth/deps/ebsynth/bin/ebsynth.app' | |
| else: | |
| print('Cannot recognize OS. Ebsynth installation stopped.') | |
| return | |
| os.chdir('src/ebsynth/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, 'model') | |
| def download_egnet_ckpt(): | |
| url = ('https://huggingface.co/PKUWilliamYang/Rerender/' | |
| 'resolve/main/models/epoch_resnet.pth') | |
| download(url, 'model') | |
| def download_hed_ckpt(): | |
| url = ('https://huggingface.co/lllyasviel/Annotators/' | |
| 'resolve/main/ControlNetHED.pth') | |
| download(url, 'src/ControlNet/annotator/ckpts') | |
| def download_depth_ckpt(): | |
| url = ('https://huggingface.co/lllyasviel/ControlNet/' | |
| 'resolve/main/annotator/ckpts/dpt_hybrid-midas-501f0c75.pt') | |
| download(url, 'src/ControlNet/annotator/ckpts') | |
| def download_ebsynth_ckpt(): | |
| os_str = platform.system() | |
| if os_str == 'Linux': | |
| url = ('https://huggingface.co/PKUWilliamYang/Rerender/' | |
| 'resolve/main/models/ebsynth') | |
| download(url, 'src/ebsynth/deps/ebsynth/bin') | |
| elif os_str == 'Windows': | |
| url = ('https://huggingface.co/PKUWilliamYang/Rerender/' | |
| 'resolve/main/models/ebsynth.exe') | |
| download(url, 'src/ebsynth/deps/ebsynth/bin') | |
| url = ('https://huggingface.co/PKUWilliamYang/Rerender/' | |
| 'resolve/main/models/ebsynth_cpu.dll') | |
| download(url, 'src/ebsynth/deps/ebsynth/bin') | |
| url = ('https://huggingface.co/PKUWilliamYang/Rerender/' | |
| 'resolve/main/models/ebsynth_cpu.exe') | |
| download(url, 'src/ebsynth/deps/ebsynth/bin') | |
| else: | |
| print('No available compiled Ebsynth.') | |
| #build_ebsynth() | |
| download_ebsynth_ckpt() | |
| download_gmflow_ckpt() | |
| download_egnet_ckpt() | |
| download_hed_ckpt() | |
| download_depth_ckpt() | |