diff --git a/llms/mlx_lm/utils.py b/llms/mlx_lm/utils.py index 8e48ab25..96be6d29 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