Upload AtriVLM

config.json

@@ -1,13 +1,9 @@
 {
-  "_name_or_path": "results/checkpoint-15818",
+  "_name_or_path": "results/IT/checkpoint-21351",
   "adjust_embedding_len": 128260,
   "architectures": [
     "AtriVLM"
   ],
-  "auto_map": {
-    "AutoConfig": "configuration_llama.VLMConfig",
-    "AutoModelForCausalLM": "modeling_VLM.AtriVLM"
-  },
   "attention_bias": false,
   "attention_dropout": 0.0,
   "bos_token_id": 128000,
@@ -42,7 +38,7 @@
   "num_hidden_layers": 16,
   "num_key_value_heads": 8,
   "num_patches": 64,
-  "pretrained_model": "results/checkpoint-15818",
+  "pretrained_model": "results/IT/checkpoint-21351",
   "pretrained_vision_model": "openai/clip-vit-large-patch14-336",
   "pretraining_tp": 1,
   "rms_norm_eps": 1e-05,
@@ -75,7 +71,7 @@
   "task_type": "CAUSAL_LM",
   "tie_word_embeddings": true,
   "torch_dtype": "float16",
-  "transformers_version": "4.46.1",
+  "transformers_version": "4.46.3",
   "use_cache": true,
   "visual_config": {
     "_attn_implementation_autoset": true,

generation_config.json

@@ -8,5 +8,5 @@
   ],
   "temperature": 0.6,
   "top_p": 0.9,
-  "transformers_version": "4.46.1"
+  "transformers_version": "4.46.3"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fc722cb8c2243ca1e96a4a9024209e81251f95b667fd49fa0d87622de0201d12
-size 3148589970
+oid sha256:bb8767c4f83358b980f889459d61be1c456bcbeb6c7e0576bd4fb5dbe88a5135
+size 3145837250