From 1afe3f2bf6ba968a9a738c2aae1ffe4a486be9d9 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 18 Feb 2025 00:13:26 -0800 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index da73ec49f6..29773d0a8e 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -128,6 +128,8 @@ class Unsloth{RLConfig_name}({RLConfig_name}): ): {RLConfig_extra_args} super().__init__({RLConfig_call_args}{RLConfig_kwargs}) + self.vllm_sampling_params = vllm_sampling_params + self.unsloth_num_chunks = unsloth_num_chunks pass {RLTrainer_extras}