apunkt/doc-to-lora
1
0
Fork 0
mirror of https://github.com/SakanaAI/doc-to-lora.git synced 2026-07-23 17:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix repeat logic

Browse source
This commit is contained in:
51616 2025-01-08 17:14:19 +00:00
parent 2381b6a93e
commit 2eb1f82371
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
hyperlora/data_utils.py
View file

@ -24,8 +24,8 @@ def filter_long_samples(samples):
def add_repeat_prompt_fn(samples):
responses = samples["answer"] + samples["context"]
ctxs = samples["context"] * 2
responses = samples["answer"] * 2
prompts = samples["prompt"]
prompts += ["Repeat the text above."] * len(prompts)