From ee31eb03e9976d159c05f1b5eecce205e65b02a1 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 11 Feb 2025 01:36:57 -0800 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index 112ba5d70a..81e929aaca 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -316,9 +316,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"): pass -def patch_vllm(trainer_file, RLTrainer_name, all_imports, imports): - import trl.trainer - RLTrainer = eval(f"trl.trainer.{trainer_file}.{RLTrainer_name}") +def patch_vllm(RLTrainer, trainer_file, RLTrainer_name, all_imports, imports): init = inspect.getsource(RLTrainer.__init__) old_init = init