.gitattributes

@@ -33,5 +33,3 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
-pytorch_model-00001-of-00002.bin filter=lfs diff=lfs merge=lfs -text
-pytorch_model-00002-of-00002.bin filter=lfs diff=lfs merge=lfs -text

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/mnt/data/zhangzheng/data/AtoM-7B/checkpoint-56000",
+  "_name_or_path": "/home/user/data/data/base_model/AtoM-7B",
   "architectures": [
     "LlamaForCausalLM"
   ],
@@ -20,8 +20,8 @@
   "rms_norm_eps": 1e-05,
   "rope_scaling": null,
   "tie_word_embeddings": false,
-  "torch_dtype": "float16",
-  "transformers_version": "4.31.0",
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.29.0",
   "use_cache": true,
   "vocab_size": 65000
 }

configuration.json

@@ -0,0 +1,11 @@
+{
+    "framework": "pytorch",
+    "task": "text-generation",
+    "model": {
+        "type": "Atom-13B"
+    },
+    "pipeline": {
+        "type": "Atom-7B-pipe"
+    },
+    "allow_remote": true
+}

generation_config.json

@@ -4,5 +4,5 @@
   "eos_token_id": 2,
   "max_length": 4096,
   "pad_token_id": 2,
-  "transformers_version": "4.31.0"
+  "transformers_version": "4.29.0"
 }

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:044f6db6c73b770408cb960c3848549c987f6612d47a71b85fe9d9c4ba499bba
+size 14017614301