Doubiiu commited on
Commit
fe012d8
·
1 Parent(s): c37e461
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,11 +23,11 @@ def download_model():
23
  REPO_ID = 'Doubiiu/DynamiCrafter'
24
  filename_list = ['model.ckpt']
25
  if not os.path.exists('./checkpoints/dynamicrafter_256_v1/'):
26
- os.makedirs('./dynamicrafter_256_v1/')
27
  for filename in filename_list:
28
  local_file = os.path.join('./checkpoints/dynamicrafter_256_v1/', filename)
29
  if not os.path.exists(local_file):
30
- hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir='./checkpoints/dynamicrafter_256_v1/', local_dir_use_symlinks=False)
31
 
32
 
33
  def infer(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123):
 
23
  REPO_ID = 'Doubiiu/DynamiCrafter'
24
  filename_list = ['model.ckpt']
25
  if not os.path.exists('./checkpoints/dynamicrafter_256_v1/'):
26
+ os.makedirs('./checkpoints/dynamicrafter_256_v1/')
27
  for filename in filename_list:
28
  local_file = os.path.join('./checkpoints/dynamicrafter_256_v1/', filename)
29
  if not os.path.exists(local_file):
30
+ hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir='./checkpoints/dynamicrafter_256_v1/', force_download=True)
31
 
32
 
33
  def infer(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123):