From 2eb1f823717b4532f5fb7170477a9ae1976be2cd Mon Sep 17 00:00:00 2001 From: 51616 Date: Wed, 8 Jan 2025 17:14:19 +0000 Subject: [PATCH] fix repeat logic --- hyperlora/data_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hyperlora/data_utils.py b/hyperlora/data_utils.py index a2e235c..b41f033 100644 --- a/hyperlora/data_utils.py +++ b/hyperlora/data_utils.py @@ -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)