From 71aff2d7f9bcbd80f798da33627a3013506d8c27 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Tue, 11 Feb 2025 03:04:05 -0800
Subject: [PATCH] Update rl.py

---
 unsloth/models/rl.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py
index 3682c71d7c..b593816409 100644
--- a/unsloth/models/rl.py
+++ b/unsloth/models/rl.py
@@ -85,10 +85,12 @@ class Unsloth{RLConfig_name}({RLConfig_name}):
         metadata = {{'help': 'vLLM SamplingParams'}},
     )
     def __init__({RLConfig_arguments},
-        sampling_params = None
+        sampling_params = None,
+        *args, **kwargs,
     ):
 {RLConfig_extra_args}
-        super().__init__({RLConfig_call_args})
+        super().__init__({RLConfig_call_args},
+        *args, **kwargs)
 pass
 
 {RLTrainer_extras}
@@ -97,11 +99,13 @@ class Unsloth{RLTrainer_name}(_Unsloth{RLTrainer_name}):
     """
     {__RLTrainer_doc__}
     """
-    def __init__({RLTrainer_arguments}
+    def __init__({RLTrainer_arguments},
+        *args, **kwargs,
     ):
         if args is None: args = Unsloth{RLConfig_name}()
 {RLTrainer_extra_args}
-        super().__init__({RLTrainer_call_args})
+        super().__init__({RLTrainer_call_args},
+        *args, **kwargs)
 pass
 '''