Fix

EasyLM/models/llama/convert_easylm_to_hf.py

@@ -227,6 +227,7 @@ def write_model(loaded, model_path, model_size):
     write_json(index_dict, os.path.join(tmp_model_path, "pytorch_model.bin.index.json"))
 
     config = LlamaConfig(
+        vocab_size=params["vocab_size"],
         hidden_size=dim,
         intermediate_size=params["intermediate_size"],
         num_attention_heads=params["n_heads"],