Update config.json

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/home/work/workspace/checkpoints/tinyllava-phi2-siglip-3.1B",
+  "_name_or_path": "tinyllava/TinyLLaVA-Phi-2-SigLIP-3.1B",
   "architectures": [
     "TinyLlavaForConditionalGeneration"
   ],
@@ -13,14 +13,14 @@
   "ignore_index": -100,
   "image_aspect_ratio": "square",
   "image_token_index": -200,
-  "llm_model_name_or_path": "/home/work/workspace/checkpoints/phi-2",
+  "llm_model_name_or_path": "microsoft/phi-2",
   "model_type": "tinyllava",
   "num_queries": 128,
   "num_resampler_layers": 3,
   "pad_token": "<|endoftext|>",
   "resampler_hidden_size": 768,
   "text_config": {
-    "_name_or_path": "/home/work/workspace/checkpoints/phi-2",
+    "_name_or_path": "microsoft/phi-2",
     "architectures": [
       "PhiForCausalLM"
     ],
@@ -44,7 +44,7 @@
     "vocab_size": 51200
   },
   "tokenizer_model_max_length": 3072,
-  "tokenizer_name_or_path": "/home/work/workspace/checkpoints/phi-2",
+  "tokenizer_name_or_path": "microsoft/phi-2",
   "tokenizer_padding_side": "right",
   "tokenizer_use_fast": false,
   "torch_dtype": "float16",
@@ -60,7 +60,7 @@
     "image_size": 384,
     "intermediate_size": 4304,
     "layer_norm_eps": 1e-06,
-    "model_name_or_path": "/home/work/workspace/checkpoints/siglip",
+    "model_name_or_path": "google/siglip-so400m-patch14-384",
     "model_name_or_path2": "",
     "model_type": "siglip_vision_model",
     "num_attention_heads": 16,
@@ -70,7 +70,7 @@
   "vision_feature_layer": -2,
   "vision_feature_select_strategy": "patch",
   "vision_hidden_size": 1152,
-  "vision_model_name_or_path": "/home/work/workspace/checkpoints/siglip",
+  "vision_model_name_or_path": "google/siglip-so400m-patch14-384",
   "vision_model_name_or_path2": "",
   "vocab_size": 51200
 }