Create download_model.py

download_model.py

@@ -0,0 +1,11 @@
+# download_model.py
+import torch
+from transformers import AutoModel, AutoTokenizer
+
+model_id = "OpenGVLab/InternVL2_5-2B"
+
+print(f"Downloading {model_id}...")
+# Download model and tokenizer to the default cache directory
+model = AutoModel.from_pretrained(model_id, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
+print("Download complete.")