[tiny] Remove use_liger argument (#1779)

Author: wizeng23

src/oumi/core/configs/training_config.py

@@ -132,13 +132,13 @@ def __post_init__(self):
 
         # Set Liger kernel flags if using a HF trainer, and if so, don't do Liger
         # patch ourselves.
-        # TODO(OPE-1117): Clean up this logic after upgrading to trl 0.16.
         if self.model.enable_liger_kernel:
-            if trainer_type == TrainerType.TRL_SFT:
-                self.training.trainer_kwargs["use_liger"] = True
-                self.training.trainer_kwargs["use_liger_kernel"] = True
-                self.model.enable_liger_kernel = False
-            elif trainer_type in (TrainerType.TRL_DPO, TrainerType.HF):
+            if trainer_type in (
+                TrainerType.TRL_SFT,
+                TrainerType.TRL_DPO,
+                TrainerType.TRL_GRPO,
+                TrainerType.HF,
+            ):
                 self.training.trainer_kwargs["use_liger_kernel"] = True
                 self.model.enable_liger_kernel = False
             elif trainer_type == TrainerType.OUMI:

tests/e2e/test_train_e2e.py

@@ -446,6 +446,20 @@ def test_train_text_1gpu_24gb(
             save_steps=5,
             is_lora=True,
         ),
+        TrainTestConfig(
+            test_name="train_text_llama3_1_8b_trl_sft_full",
+            config_path=(
+                get_configs_dir()
+                / "recipes"
+                / "llama3_1"
+                / "sft"
+                / "8b_full"
+                / "train.yaml"
+            ),
+            trainer_type=TrainerType.TRL_SFT,
+            max_steps=5,
+            save_steps=5,
+        ),
     ],
     ids=get_train_test_id_fn,
 )