justshao
/

llama2-7b-with-confidence

Text Classification

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

justshao commited on Apr 22

Commit

8a93c2e

•

1 Parent(s): b0d9ffc

Upload config

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -16,6 +16,7 @@
   "num_attention_heads": 32,
   "num_hidden_layers": 32,
   "num_key_value_heads": 32,
   "pretraining_tp": 1,
   "rms_norm_eps": 1e-05,
   "rope_scaling": null,
@@ -24,5 +25,5 @@
   "torch_dtype": "bfloat16",
   "transformers_version": "4.39.3",
   "use_cache": true,
-  "vocab_size": 32000
 }

   "num_attention_heads": 32,
   "num_hidden_layers": 32,
   "num_key_value_heads": 32,
+  "pad_token_id": 32000,
   "pretraining_tp": 1,
   "rms_norm_eps": 1e-05,
   "rope_scaling": null,
   "torch_dtype": "bfloat16",
   "transformers_version": "4.39.3",
   "use_cache": true,
+  "vocab_size": 32001
 }