From f27221203863ba19f4d31c6a045ad6dc9ade85e0 Mon Sep 17 00:00:00 2001 From: xrdaukar Date: Thu, 30 Jan 2025 16:25:07 -0800 Subject: [PATCH] Fix typo and update warning message for OUMI trainer (#1319) --- src/oumi/builders/training.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/oumi/builders/training.py b/src/oumi/builders/training.py index 58c0d416b..53a43e896 100644 --- 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()