diff --git a/llms/mlx_lm/utils.py b/llms/mlx_lm/utils.py index 96be6d29..68d2204e 100644 --- a/llms/mlx_lm/utils.py +++ b/llms/mlx_lm/utils.py @@ -43,7 +43,7 @@ MODEL_REMAPPING = { "mistral": "llama", # mistral is compatible with llama "phi-msft": "phixtral", "falcon_mamba": "mamba", - "qwen2": "qwen2", + "qwen2": "qwen2" } MAX_FILE_SIZE_GB = 5