zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-24 17:31:18 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats 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 from mlx.utils import tree_flatten, tree_map, tree_unflatten
def convert(weights, config): def convert(tf, config):
def convert_single(k, v): def convert_single(k, v):
v = v.to(torch.float16).numpy() v = v.to(torch.float16).numpy()
if "block_sparse_moe" not in k: if "block_sparse_moe" not in k: