chargoddard
/

internlm2-7b-llama

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

chargoddard commited on Jan 18

Commit

67517f8

•

1 Parent(s): 4617ee7

Update convert_weights.py

Files changed (1) hide show

convert_weights.py +1 -1

convert_weights.py CHANGED Viewed

@@ -84,7 +84,7 @@ writer.finalize()
 cfg_dict = json.loads(cfg.to_json_string())
 del cfg_dict["auto_map"]
-cfg_dict["architectures"] = "LlamaForCausalLM"
 cfg_dict["model_type"] = "llama"
 if "rope_scaling" in cfg_dict and cfg_dict["rope_scaling"]["factor"] == 1.0:
     del cfg_dict["rope_scaling"]

 cfg_dict = json.loads(cfg.to_json_string())
 del cfg_dict["auto_map"]
+cfg_dict["architectures"] = ["LlamaForCausalLM"]
 cfg_dict["model_type"] = "llama"
 if "rope_scaling" in cfg_dict and cfg_dict["rope_scaling"]["factor"] == 1.0:
     del cfg_dict["rope_scaling"]