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

fix bad convert parameter (#178)

Browse Source
This commit is contained in:
Daniel Strobusch
2023-12-23 16:09:49 +01:00
committed by GitHub
parent f4709cb807
commit 092e87211e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
llms/mixtral/convert.py
View File

@@ -15,7 +15,7 @@ from mixtral import Mixtral, ModelArgs
from mlx.utils import tree_flatten, tree_map, tree_unflatten
def convert(weights, config):
def convert(tf, config):
def convert_single(k, v):
v = v.to(torch.float16).numpy()
if "block_sparse_moe" not in k: