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

Update __init__.py

Browse source
This commit is contained in:
Daniel Han 2024-12-29 19:36:11 -08:00
commit 9611ee433e
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
unsloth/__init__.py
View file

@ -106,12 +106,12 @@ try:
text = f.read()
# See https://github.com/facebookresearch/xformers/issues/1176#issuecomment-2545829591
if "num_splits_key=-1," in text:
print("Unsloth: Patching Xformers to fix some performance issues.")
text = text.replace("num_splits_key=-1,", "num_splits_key=None,")
f.seek(0)
f.write(text)
f.truncate()
print("Unsloth: Patching Xformers to fix some performance issues.")
pass
f.seek(0)
f.write(text)
f.truncate()
pass
pass
pass