From 154457dbf75b749a908e545e9a4945dda7665462 Mon Sep 17 00:00:00 2001 From: Long Yixing Date: Wed, 10 Jun 2026 22:55:50 +0800 Subject: [PATCH] fix(studio): reject unsupported mlx training modes (#6091) --- studio/backend/core/training/worker.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/studio/backend/core/training/worker.py b/studio/backend/core/training/worker.py index 0a3b1f2019..ff4329ee13 100644 --- a/studio/backend/core/training/worker.py +++ b/studio/backend/core/training/worker.py @@ -1373,6 +1373,14 @@ def _run_mlx_training(event_queue, stop_queue, config): message = "LoftQ is not supported for MLX training yet." _send("error", error = message) raise NotImplementedError(message) + if config.get("is_embedding"): + message = "Embedding model training is not supported for MLX training yet." + _send("error", error = message) + raise NotImplementedError(message) + if config.get("training_type") == "Continued Pretraining": + message = "Continued Pretraining is not supported for MLX training yet." + _send("error", error = message) + raise NotImplementedError(message) optim_name = _normalize_mlx_studio_optimizer(config.get("optim", "adamw_8bit")) lr_scheduler_type = _normalize_mlx_studio_scheduler(config.get("lr_scheduler_type", "linear"))