Update rl_replacements.py
This commit is contained in:
parent
36e1bc35f0
commit
f34fec15e6
1 changed files with 2 additions and 0 deletions
|
|
@ -90,6 +90,8 @@ def sft_trainer_prepare_dataset(function_name, function):
|
|||
if matched:
|
||||
# Use fast version!
|
||||
function = inspect.getsource(fast_sft_prepare_dataset)
|
||||
function = function.split("\n")
|
||||
function = "\n".join(" "*4 + x for x in function)
|
||||
function = function.replace("def sft_prepare_dataset", "def _prepare_dataset")
|
||||
return function
|
||||
pass
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue