|
import os |
|
import requests |
|
import time |
|
import gdown |
|
import sys |
|
from huggingface_hub import snapshot_download |
|
|
|
def download_model(): |
|
if not os.path.exists(CheckpointsDir): |
|
os.makedirs(CheckpointsDir) |
|
print("Checkpoint Not Downloaded, start downloading...") |
|
tic = time.time() |
|
snapshot_download( |
|
repo_id="TMElyralab/MuseTalk", |
|
local_dir=CheckpointsDir, |
|
max_workers=8, |
|
local_dir_use_symlinks=True, |
|
) |
|
|
|
snapshot_download( |
|
repo_id="stabilityai/sd-vae-ft-mse", |
|
local_dir=CheckpointsDir, |
|
max_workers=8, |
|
local_dir_use_symlinks=True, |
|
) |
|
|
|
snapshot_download( |
|
repo_id="yzd-v/DWPose", |
|
local_dir=CheckpointsDir, |
|
max_workers=8, |
|
local_dir_use_symlinks=True, |
|
) |
|
|
|
url = "https://openaipublic.azureedge.net/main/whisper/models/65147644a518d12f04e32d6f3b26facc3f8dd46e5390956a9424a650c0ce22b9/tiny.pt" |
|
response = requests.get(url) |
|
|
|
if response.status_code == 200: |
|
|
|
file_path = f"{CheckpointsDir}/whisper/tiny.pt" |
|
os.makedirs(f"{CheckpointsDir}/whisper/") |
|
|
|
with open(file_path, "wb") as f: |
|
f.write(response.content) |
|
else: |
|
print(f"请求失败,状态码:{response.status_code}") |
|
|
|
url = "https://drive.google.com/uc?id=154JgKpzCPW82qINcVieuPH3fZ2e0P812" |
|
os.makedirs(f"{CheckpointsDir}/face-parse-bisent/") |
|
file_path = f"{CheckpointsDir}/face-parse-bisent/79999_iter.pth" |
|
gdown.download(url, file_path, quiet=False) |
|
|
|
url = "https://download.pytorch.org/models/resnet18-5c106cde.pth" |
|
response = requests.get(url) |
|
|
|
if response.status_code == 200: |
|
|
|
file_path = f"{CheckpointsDir}/face-parse-bisent/resnet18-5c106cde.pth" |
|
|
|
with open(file_path, "wb") as f: |
|
f.write(response.content) |
|
else: |
|
print(f"请求失败,状态码:{response.status_code}") |
|
|
|
|
|
toc = time.time() |
|
|
|
print(f"download cost {toc-tic} seconds") |
|
else: |
|
print("Already download the model.") |
|
|
|
|
|
ProjectDir = os.path.abspath(os.path.dirname(__file__)) |
|
CheckpointsDir = os.path.join(ProjectDir, "checkpoints") |
|
|
|
download_model() |