Minor adjustment

h2oai · Mar 30, 2024 · 40af567 · 40af567
1 parent e330cc6
commit 40af567
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/gen.py b/src/gen.py
@@ -5875,7 +5875,8 @@ def get_limited_prompt(instruction,
     if use_chat_template:
         context2 = apply_chat_template(instruction, system_prompt, history, tokenizer)
         iinput = ''
-        context = ''
+        context1 = ''
+        num_context1_tokens = 0
     else:
         context2 = history_to_context_func(history)