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

Update _utils.py

Browse source
This commit is contained in:
Daniel Han 2024-11-05 22:48:36 -08:00
commit bdfb2b993a
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/models/_utils.py
View file

@ -396,8 +396,8 @@ patch_torch_compile(
)
torch_compile_options = {
"epilogue_fusion" : True,
"max_autotune" : True,
"epilogue_fusion" : False,
"max_autotune" : False,
"shape_padding" : True,
"trace.enabled" : UNSLOTH_COMPILE_DEBUG,
"triton.cudagraphs" : False,