ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update rl.py

Browse source
This commit is contained in:
Daniel Han 2025-02-11 03:04:05 -08:00
commit 71aff2d7f9
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
unsloth/models/rl.py
View file

@ -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
'''