Update app.py

add force_download to make env build more stable
This commit is contained in:
Tuuz
2024-04-16 19:32:25 +08:00
committed by GitHub
parent 696ec5aa03
commit ae956e63e0

3
app.py
View File

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