AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add back a missing bos token

Browse Source
This commit is contained in:
oobabooga 2023-05-24 13:54:36 -03:00
parent 3cd7c5bdd0
commit 63ce5f9c28
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/training.py
View File

@ -262,7 +262,7 @@ def do_train(lora_name: str, always_override: bool, save_steps: int, micro_batch
else:
ind = prompt.index(train_only_after) + len(train_only_after)
before_tokens = encode(prompt[:ind], False)
before_tokens = encode(prompt[:ind], True)
after_tokens = encode(prompt[ind:], False)
full_length = len(after_tokens) + len(before_tokens)