Skip to content

Commit ea3a08e

Browse files
authored
Fix empty context (#165)
Signed-off-by: SimJeg <sjegou@nvidia.com>
1 parent 8306602 commit ea3a08e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

kvpress/pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ def preprocess(
134134
context = bos_token + context
135135
question_suffix = "\n" # to separate the question from the answer
136136
else:
137-
separator = "\n" + "#" * len(context)
137+
separator = "#" * (len(context) + 10)
138138
context = self.tokenizer.apply_chat_template(
139139
[{"role": "user", "content": context + separator}],
140140
add_generation_prompt=True,

0 commit comments

Comments
 (0)