AI/alpaca-lora
1
0
Fork 0
mirror of https://github.com/tloen/alpaca-lora.git synced 2024-10-01 01:05:56 -04:00
Code Issues Packages Projects Releases Wiki Activity

lint

Browse Source
This commit is contained in:
Eric Wang 2023-04-13 13:52:17 -07:00
parent ea0e96b017
commit 65fb8225c0
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
finetune.py
View File

@ -156,7 +156,9 @@ def train(
user_prompt = prompter.generate_prompt( user_prompt = prompter.generate_prompt(
data_point["instruction"], data_point["input"] data_point["instruction"], data_point["input"]
) )
tokenized_user_prompt = tokenize(user_prompt, add_eos_token=add_eos_token) tokenized_user_prompt = tokenize(
user_prompt, add_eos_token=add_eos_token
)
user_prompt_len = len(tokenized_user_prompt["input_ids"]) user_prompt_len = len(tokenized_user_prompt["input_ids"])
if add_eos_token: if add_eos_token: