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-14 04:35:03 -08:00
commit 4d8cc00592
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/models/rl.py
View file

@ -109,13 +109,13 @@ from packaging.version import Version
import torch
from contextlib import nullcontext
from torch.nn import functional as F
torch_compile_options = {
torch_compile_options = {{
"epilogue_fusion" : True,
"max_autotune" : True,
"shape_padding" : True,
"trace.enabled" : False,
"triton.cudagraphs" : False,
}
}}
{selective_log_softmax_code}
{RL_pre}