zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-10-31 19:18:09 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #107 from ml-explore/hf_mixtral

Browse Source 
Use official HF for mixtral
This commit is contained in:
Awni Hannun
2023-12-14 16:57:19 -08:00
committed by GitHub
parent 67a208b13e b863e7cca0
commit a3ecda22fe
11 changed files with 106 additions and 32 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"