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

Update llama.py

Browse source
This commit is contained in:
Daniel Han 2024-12-01 02:22:21 -08:00
commit 4620e76e3d
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
unsloth/models/llama.py
View file

@ -1736,11 +1736,11 @@ class FastLlamaModel:
print(debug_info)
debug_info = debug_info\
.replace("{unsloth_0}", re.escape(unsloth_0))\
.replace("{unsloth_1}", re.escape(unsloth_1))\
.replace("{unsloth_2}", re.escape(unsloth_2))\
.replace("{unsloth_3}", re.escape(unsloth_3))\
.replace("{unsloth_4}", re.escape(unsloth_4))
.replace("{unsloth_0}", re.unescape(unsloth_0))\
.replace("{unsloth_1}", re.unescape(unsloth_1))\
.replace("{unsloth_2}", re.unescape(unsloth_2))\
.replace("{unsloth_3}", re.unescape(unsloth_3))\
.replace("{unsloth_4}", re.unescape(unsloth_4))
debug_info = debug_info.split('\n')
debug_info = "\n".join([debug_info[0]] + [spaces + x[8:] for x in debug_info[1:]])