From a87efe86da5754a013fcc01d0d1f7cf1126a7dec Mon Sep 17 00:00:00 2001 From: Dillon DuPont Date: Mon, 28 Apr 2025 14:34:59 -0400 Subject: [PATCH] correct mappings --- libs/agent/agent/ui/gradio/app.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/agent/agent/ui/gradio/app.py b/libs/agent/agent/ui/gradio/app.py index 16c5f7e6..354580d7 100644 --- a/libs/agent/agent/ui/gradio/app.py +++ b/libs/agent/agent/ui/gradio/app.py @@ -165,8 +165,8 @@ MODEL_MAPPINGS = { "uitars": { # UI-TARS models using MLXVLM provider "default": "mlx-community/UI-TARS-1.5-7B-4bit", - "UI-TARS-1.5-7B-4bit": "mlx-community/UI-TARS-1.5-7B-4bit", - "UI-TARS-1.5-7B-6bit": "mlx-community/UI-TARS-1.5-7B-6bit" + "mlx-community/UI-TARS-1.5-7B-4bit": "mlx-community/UI-TARS-1.5-7B-4bit", + "mlx-community/UI-TARS-1.5-7B-6bit": "mlx-community/UI-TARS-1.5-7B-6bit" }, "ollama": { # For Ollama models, we keep the original name