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}