zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-30 02:53:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix conversion

Browse Source
This commit is contained in:
Awni Hannun 2023-12-10 14:35:39 -08:00 committed by Angelos Katharopoulos
parent 8e801fcc57
commit cba703d1f1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lora/convert.py
View File

@ -47,6 +47,7 @@ if __name__ == "__main__":
# Copy the params
with open(torch_path / "params.json", "r") as f:
config = json.loads(f.read())
n_heads = config["n_heads"]
if "sliding_window" in config:
config.pop("sliding_window")
if "n_kv_heads" not in config: