Spaces:
Runtime error
Runtime error
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)
|