Spaces:
Paused
Paused
Delete download_model.py with huggingface_hub
Browse files- download_model.py +0 -11
download_model.py
DELETED
|
@@ -1,11 +0,0 @@
|
|
| 1 |
-
# download_model.py
|
| 2 |
-
import torch
|
| 3 |
-
from transformers import AutoModel, AutoTokenizer
|
| 4 |
-
|
| 5 |
-
model_id = "OpenGVLab/InternVL2_5-2B"
|
| 6 |
-
|
| 7 |
-
print(f"Downloading {model_id}...")
|
| 8 |
-
# Download model and tokenizer to the default cache directory
|
| 9 |
-
model = AutoModel.from_pretrained(model_id, trust_remote_code=True)
|
| 10 |
-
tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
|
| 11 |
-
print("Download complete.")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|