HuggingFaceM4
/

tiny-random-LlamaForCausalLM

Text Generation

text-generation-inference

Model card Files Files and versions

Match the tokenizer

#5

by lsb - opened 15 days ago

base: refs/heads/main

←

from: refs/pr/5

Discussion Files changed

Files changed (1) hide show

config.json +3 -3

config.json CHANGED Viewed

@@ -2,8 +2,8 @@
   "architectures": [
     "LlamaForCausalLM"
   ],
-  "bos_token_id": 0,
-  "eos_token_id": 1,
   "hidden_act": "silu",
   "hidden_size": 16,
   "initializer_range": 0.02,
@@ -11,7 +11,7 @@
   "model_type": "llama",
   "num_attention_heads": 4,
   "num_hidden_layers": 2,
-  "pad_token_id": -1,
   "rms_norm_eps": 1e-06,
   "tie_word_embeddings": false,
   "torch_dtype": "float16",

   "architectures": [
     "LlamaForCausalLM"
   ],
+  "bos_token_id": 1,
+  "eos_token_id": 2,
   "hidden_act": "silu",
   "hidden_size": 16,
   "initializer_range": 0.02,
   "model_type": "llama",
   "num_attention_heads": 4,
   "num_hidden_layers": 2,
+  "pad_token_id": 0,
   "rms_norm_eps": 1e-06,
   "tie_word_embeddings": false,
   "torch_dtype": "float16",