From c355a7457d956ca32cf37bb226248d0df01c58b6 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 17 Feb 2025 22:24:57 -0800 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index fa617d5d46..231dbe7765 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -122,7 +122,8 @@ class Unsloth{RLConfig_name}({RLConfig_name}): metadata = {{'help': 'Chunk size to reduce memory usage'}}, ) def __init__({RLConfig_arguments}, - vllm_sampling_params = None, + vllm_sampling_params = vllm_sampling_params, + unsloth_num_chunks = unsloth_num_chunks, **kwargs, ): {RLConfig_extra_args}