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

Merge ec06c04f4f into e8afb59de4

Browse Source
This commit is contained in:
Sindhu Satish 2025-01-29 07:42:19 -08:00 committed by GitHub
commit 70360d9d0d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llms/mlx_lm/lora.py
View File

@ -296,4 +296,4 @@ def main():
if __name__ == "__main__": if __name__ == "__main__":
main() main()

3
llms/mlx_lm/utils.py
View File

@ -43,6 +43,7 @@ MODEL_REMAPPING = {
"mistral": "llama", # mistral is compatible with llama "mistral": "llama", # mistral is compatible with llama
"phi-msft": "phixtral", "phi-msft": "phixtral",
"falcon_mamba": "mamba", "falcon_mamba": "mamba",
"qwen2": "qwen2"
} }
MAX_FILE_SIZE_GB = 5 MAX_FILE_SIZE_GB = 5
@ -1026,4 +1027,4 @@ def convert(
save_config(config, config_path=mlx_path / "config.json") save_config(config, config_path=mlx_path / "config.json")
if upload_repo is not None: if upload_repo is not None:
upload_to_hub(mlx_path, upload_repo, hf_path) upload_to_hub(mlx_path, upload_repo, hf_path)