Update rl.py

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}