wondervictor commited on
Commit
2ab105b
·
verified ·
1 Parent(s): f3e5467

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -33,6 +33,7 @@ print("Torch version:", torch.__version__)
33
  ckpt_folder = './checkpoints'
34
  t5_folder = os.path.join(ckpt_folder, "flan-t5-xl/flan-t5-xl")
35
  dinov2_folder = os.path.join(ckpt_folder, "dinov2-small")
 
36
  hf_hub_download(repo_id="google/flan-t5-xl", filename="config.json", local_dir=t5_folder)
37
  hf_hub_download(repo_id="google/flan-t5-xl", filename="pytorch_model-00001-of-00002.bin", local_dir=t5_folder)
38
  hf_hub_download(repo_id="google/flan-t5-xl", filename="pytorch_model-00002-of-00002.bin", local_dir=t5_folder)
 
33
  ckpt_folder = './checkpoints'
34
  t5_folder = os.path.join(ckpt_folder, "flan-t5-xl/flan-t5-xl")
35
  dinov2_folder = os.path.join(ckpt_folder, "dinov2-small")
36
+ dinov2_folder = os.path.join(ckpt_folder, "dinov2-base")
37
  hf_hub_download(repo_id="google/flan-t5-xl", filename="config.json", local_dir=t5_folder)
38
  hf_hub_download(repo_id="google/flan-t5-xl", filename="pytorch_model-00001-of-00002.bin", local_dir=t5_folder)
39
  hf_hub_download(repo_id="google/flan-t5-xl", filename="pytorch_model-00002-of-00002.bin", local_dir=t5_folder)