Commit
•
67517f8
1
Parent(s):
4617ee7
Update convert_weights.py
Browse files- convert_weights.py +1 -1
convert_weights.py
CHANGED
|
@@ -84,7 +84,7 @@ writer.finalize()
|
|
| 84 |
|
| 85 |
cfg_dict = json.loads(cfg.to_json_string())
|
| 86 |
del cfg_dict["auto_map"]
|
| 87 |
-
cfg_dict["architectures"] = "LlamaForCausalLM"
|
| 88 |
cfg_dict["model_type"] = "llama"
|
| 89 |
if "rope_scaling" in cfg_dict and cfg_dict["rope_scaling"]["factor"] == 1.0:
|
| 90 |
del cfg_dict["rope_scaling"]
|
|
|
|
| 84 |
|
| 85 |
cfg_dict = json.loads(cfg.to_json_string())
|
| 86 |
del cfg_dict["auto_map"]
|
| 87 |
+
cfg_dict["architectures"] = ["LlamaForCausalLM"]
|
| 88 |
cfg_dict["model_type"] = "llama"
|
| 89 |
if "rope_scaling" in cfg_dict and cfg_dict["rope_scaling"]["factor"] == 1.0:
|
| 90 |
del cfg_dict["rope_scaling"]
|