Oli_Zhan commited on
Commit
f7602ff
·
unverified ·
2 Parent(s): 6b09521 2b34ec1

Merge pull request #31 from tobycroft/main

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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"