chenjianfei commited on
Commit
a0189a2
·
1 Parent(s): 65e0347
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -125,7 +125,8 @@ def predict(message, chat_history, thinking_history, config, section_state):
125
  text = core_tokenizer.apply_chat_template(
126
  input_message,
127
  tokenize=False,
128
- add_generation_prompt=True
 
129
  )
130
  model_inputs = core_tokenizer([text], return_tensors="pt").to(core_llm.device)
131
  # conduct text completion
 
125
  text = core_tokenizer.apply_chat_template(
126
  input_message,
127
  tokenize=False,
128
+ add_generation_prompt=True,
129
+ enable_thinking=False
130
  )
131
  model_inputs = core_tokenizer([text], return_tensors="pt").to(core_llm.device)
132
  # conduct text completion