From d6ae910edc1b161fb7d5622a8b159dcff67d2ef9 Mon Sep 17 00:00:00 2001 From: Roland Tannous Date: Tue, 10 Mar 2026 19:05:47 +0000 Subject: [PATCH] fix: propagate is_embedding into worker subprocess config start_training() cherry-picks kwargs into a config dict but was missing is_embedding, so config.get("is_embedding", False) in worker.py always returned False and embedding training never ran. --- studio/backend/core/training/training.py | 1 + 1 file changed, 1 insertion(+) diff --git a/studio/backend/core/training/training.py b/studio/backend/core/training/training.py index 14ff57bbd7..5f9cb86b0c 100644 --- a/studio/backend/core/training/training.py +++ b/studio/backend/core/training/training.py @@ -150,6 +150,7 @@ class TrainingBackend: "custom_format_mapping": kwargs.get("custom_format_mapping"), "is_dataset_image": kwargs.get("is_dataset_image", False), "is_dataset_audio": kwargs.get("is_dataset_audio", False), + "is_embedding": kwargs.get("is_embedding", False), "num_epochs": kwargs.get("num_epochs", 3), "learning_rate": kwargs.get("learning_rate", "2e-4"), "batch_size": kwargs.get("batch_size", 2),