Spaces:
Running
on
Zero
Running
on
Zero
wondervictor
commited on
Update app.py
Browse files
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)
|