lvwerra HF staff commited on
Commit
7eaa8e3
1 Parent(s): 5a68917

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,7 @@ def generate(prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition
94
  raise ValueError("Chat mode and FIM are mutually exclusive. Choose one or the other.")
95
 
96
  if chat_mode:
97
- chat_prompt = "Human: " + prompt + "\n\nAssistant: "
98
  prompt = HHH_PROMPT + chat_prompt
99
 
100
  if FIM_INDICATOR in prompt:
 
94
  raise ValueError("Chat mode and FIM are mutually exclusive. Choose one or the other.")
95
 
96
  if chat_mode:
97
+ chat_prompt = "Human: " + prompt + "\n\nAssistant:"
98
  prompt = HHH_PROMPT + chat_prompt
99
 
100
  if FIM_INDICATOR in prompt: