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

Auto patching

Browse source
This commit is contained in:
Daniel Han 2025-02-10 23:33:15 -08:00
commit 2aa87bd8f1
2 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/models/llama.py
View file

@ -2739,3 +2739,5 @@ class FastLlamaModel:
pass
pass
from .rl import PatchFastRL
PatchFastRL(FastLanguageModel = FastLlamaModel)

2
unsloth/models/rl.py
View file

@ -453,5 +453,5 @@ pass
def PatchFastRL(algorithm = "GRPO", FastLanguageModel = None):
if FastLanguageModel is not None: PatchRL(FastLanguageModel)
patch_trl_rl_trainers()
PatchRLStatistics(algorithm)
if algorithm is nont None: PatchRLStatistics(algorithm)
pass