Merge pull request #31 from tobycroft/main

This commit is contained in:
Oli_Zhan
2024-04-17 14:34:35 +08:00
committed by GitHub

3
app.py
View File

@@ -49,6 +49,7 @@ def download_model():
local_dir=CheckpointsDir, local_dir=CheckpointsDir,
max_workers=8, max_workers=8,
local_dir_use_symlinks=True, local_dir_use_symlinks=True,
force_download=True, resume_download=False
) )
# weight # weight
os.makedirs(f"{CheckpointsDir}/sd-vae-ft-mse/") os.makedirs(f"{CheckpointsDir}/sd-vae-ft-mse/")
@@ -57,6 +58,7 @@ def download_model():
local_dir=CheckpointsDir+'/sd-vae-ft-mse', local_dir=CheckpointsDir+'/sd-vae-ft-mse',
max_workers=8, max_workers=8,
local_dir_use_symlinks=True, local_dir_use_symlinks=True,
force_download=True, resume_download=False
) )
#dwpose #dwpose
os.makedirs(f"{CheckpointsDir}/dwpose/") os.makedirs(f"{CheckpointsDir}/dwpose/")
@@ -65,6 +67,7 @@ def download_model():
local_dir=CheckpointsDir+'/dwpose', local_dir=CheckpointsDir+'/dwpose',
max_workers=8, max_workers=8,
local_dir_use_symlinks=True, local_dir_use_symlinks=True,
force_download=True, resume_download=False
) )
#vae #vae
url = "https://openaipublic.azureedge.net/main/whisper/models/65147644a518d12f04e32d6f3b26facc3f8dd46e5390956a9424a650c0ce22b9/tiny.pt" url = "https://openaipublic.azureedge.net/main/whisper/models/65147644a518d12f04e32d6f3b26facc3f8dd46e5390956a9424a650c0ce22b9/tiny.pt"