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

Merge pull request #40 from Jacksonzhang0316/main

Browse Source 
fix: Unsupported BFloat16 Data Type Issue with MPS Backend
This commit is contained in:
Awni Hannun 2023-12-08 06:48:21 -08:00 committed by GitHub
commit 6259c9a048
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
lora/convert.py
View File

@ -32,7 +32,12 @@ def map_torch_to_mlx(key, value):
elif "rope" in key:
return None, None
return key, value.numpy()
return (
key,
value.numpy()
if value.dtype != torch.bfloat16
else value.to(torch.float32).numpy(),
)
if __name__ == "__main__":