diff --git a/llms/mlx_lm/lora.py b/llms/mlx_lm/lora.py index 43f508c3..fc004ae2 100644 --- a/llms/mlx_lm/lora.py +++ b/llms/mlx_lm/lora.py @@ -296,4 +296,4 @@ def main(): if __name__ == "__main__": - main() + main() \ No newline at end of file diff --git a/llms/mlx_lm/utils.py b/llms/mlx_lm/utils.py index 0150f1b7..6007adb0 100644 --- a/llms/mlx_lm/utils.py +++ b/llms/mlx_lm/utils.py @@ -43,6 +43,7 @@ MODEL_REMAPPING = { "mistral": "llama", # mistral is compatible with llama "phi-msft": "phixtral", "falcon_mamba": "mamba", + "qwen2": "qwen2" } MAX_FILE_SIZE_GB = 5 @@ -1026,4 +1027,4 @@ def convert( save_config(config, config_path=mlx_path / "config.json") if upload_repo is not None: - upload_to_hub(mlx_path, upload_repo, hf_path) + upload_to_hub(mlx_path, upload_repo, hf_path) \ No newline at end of file