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

format

Browse Source
This commit is contained in:
Awni Hannun
2023-12-14 16:56:50 -08:00
parent e434e7e5c2
commit b863e7cca0
8 changed files with 33 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
phi2/convert.py
View File

@@ -1,6 +1,7 @@
from transformers import AutoModelForCausalLM
import numpy as np
def replace_key(key: str) -> str:
if "wte.weight" in key:
key = "wte.weight"