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-15 18:34:25 -08:00
commit 285efeb30f
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/models/rl.py
View file

@ -622,9 +622,9 @@ def patch_functions(RLTrainer, trainer_file, RLTrainer_name, all_imports, import
pass
# Remove non editted functions
for remove in remover:
RLTrainer_source = RLTrainer_source.replace(remove, "")
pass
# for remove in remover:
# RLTrainer_source = RLTrainer_source.replace(remove, "")
# pass
RLTrainer_source = RLTrainer_source.replace(
f"class {RLTrainer_name}", f"class _Unsloth{RLTrainer_name}", 1