Upload AtriVLM

config.json

@@ -1,13 +1,9 @@
 {
-  "_name_or_path": "results/checkpoint-5000",
+  "_name_or_path": "results/checkpoint-10000",
   "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-5000",
+  "pretrained_model": "results/checkpoint-10000",
   "pretrained_vision_model": "openai/clip-vit-large-patch14-336",
   "pretraining_tp": 1,
   "rms_norm_eps": 1e-05,

generation_config.json

@@ -1,12 +1,12 @@
-{
-  "bos_token_id": 128000,
-  "do_sample": true,
-  "eos_token_id": [
-    128001,
-    128008,
-    128009
-  ],
-  "temperature": 0.6,
-  "top_p": 0.9,
-  "transformers_version": "4.46.1"
-}
+{
+  "bos_token_id": 128000,
+  "do_sample": true,
+  "eos_token_id": [
+    128001,
+    128008,
+    128009
+  ],
+  "temperature": 0.6,
+  "top_p": 0.9,
+  "transformers_version": "4.46.1"
+}

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:197c6fdc090a5bce6c2821e208ad0363f1fabc16534186cd65dadf14ac2b39e1
+oid sha256:306d3795471a002162b5ba615c774668deff2808aab727aef1c027a9600174f3
 size 3148589970