From ab27ddc6a3b1f6951c28c1899f2a12e439cbba93 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}