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