convert model to bf16

config.json

@@ -1,4 +1,5 @@
 {
+  "_name_or_path": "c:/cp2024",
   "architectures": [
     "LlamaForCausalLM"
   ],
@@ -6,6 +7,7 @@
   "attention_dropout": 0.0,
   "bos_token_id": 1,
   "eos_token_id": 2,
+  "head_dim": 48,
   "hidden_act": "silu",
   "hidden_size": 384,
   "initializer_range": 0.02,
@@ -21,8 +23,8 @@
   "rope_scaling": null,
   "rope_theta": 10000.0,
   "tie_word_embeddings": true,
-  "torch_dtype": "float32",
-  "transformers_version": "4.44.2",
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.46.3",
   "use_cache": true,
   "vocab_size": 32000
 }

generation_config.json

@@ -2,5 +2,5 @@
   "_from_model_config": true,
   "bos_token_id": 1,
   "eos_token_id": 2,
-  "transformers_version": "4.44.2"
+  "transformers_version": "4.46.3"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:521ff00394cf09dbbdc00f3cc3de31d3817b02f570331f1b486e2408caf71236
-size 124683704
+oid sha256:f9490c0b00e4d0b985dd4a7d4548bfbd441c5b5091f19d8cd886e2b803e4ed19
+size 62345936