Fix typo and update warning message for OUMI trainer (#1319)

oumi-ai · Jan 31, 2025 · f272212 · f272212
1 parent b8c00d8
commit f272212
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/oumi/builders/training.py b/src/oumi/builders/training.py
@@ -111,7 +111,8 @@ def _init_oumi_trainer(*args, **kwargs) -> BaseTrainer:
         return _create_hf_builder_fn(transformers.Trainer)
     elif trainer_type == TrainerType.OUMI:
         warnings.warn(
-            "OUMI trainer is still in development model. Please use HF trainer for now."
+            "OUMI trainer is still in alpha mode. "
+            "Prefer to use HF trainer when possible."
         )
         return _create_oumi_builder_fn()