Merge pull request #31 from tobycroft/main
Browse files
app.py
CHANGED
|
@@ -49,6 +49,7 @@ def download_model():
|
|
| 49 |
local_dir=CheckpointsDir,
|
| 50 |
max_workers=8,
|
| 51 |
local_dir_use_symlinks=True,
|
|
|
|
| 52 |
)
|
| 53 |
# weight
|
| 54 |
os.makedirs(f"{CheckpointsDir}/sd-vae-ft-mse/")
|
|
@@ -57,6 +58,7 @@ def download_model():
|
|
| 57 |
local_dir=CheckpointsDir+'/sd-vae-ft-mse',
|
| 58 |
max_workers=8,
|
| 59 |
local_dir_use_symlinks=True,
|
|
|
|
| 60 |
)
|
| 61 |
#dwpose
|
| 62 |
os.makedirs(f"{CheckpointsDir}/dwpose/")
|
|
@@ -65,6 +67,7 @@ def download_model():
|
|
| 65 |
local_dir=CheckpointsDir+'/dwpose',
|
| 66 |
max_workers=8,
|
| 67 |
local_dir_use_symlinks=True,
|
|
|
|
| 68 |
)
|
| 69 |
#vae
|
| 70 |
url = "https://openaipublic.azureedge.net/main/whisper/models/65147644a518d12f04e32d6f3b26facc3f8dd46e5390956a9424a650c0ce22b9/tiny.pt"
|
|
|
|
| 49 |
local_dir=CheckpointsDir,
|
| 50 |
max_workers=8,
|
| 51 |
local_dir_use_symlinks=True,
|
| 52 |
+
force_download=True, resume_download=False
|
| 53 |
)
|
| 54 |
# weight
|
| 55 |
os.makedirs(f"{CheckpointsDir}/sd-vae-ft-mse/")
|
|
|
|
| 58 |
local_dir=CheckpointsDir+'/sd-vae-ft-mse',
|
| 59 |
max_workers=8,
|
| 60 |
local_dir_use_symlinks=True,
|
| 61 |
+
force_download=True, resume_download=False
|
| 62 |
)
|
| 63 |
#dwpose
|
| 64 |
os.makedirs(f"{CheckpointsDir}/dwpose/")
|
|
|
|
| 67 |
local_dir=CheckpointsDir+'/dwpose',
|
| 68 |
max_workers=8,
|
| 69 |
local_dir_use_symlinks=True,
|
| 70 |
+
force_download=True, resume_download=False
|
| 71 |
)
|
| 72 |
#vae
|
| 73 |
url = "https://openaipublic.azureedge.net/main/whisper/models/65147644a518d12f04e32d6f3b26facc3f8dd46e5390956a9424a650c0ce22b9/tiny.pt"
|