chenjianfei
commited on
Commit
·
a0189a2
1
Parent(s):
65e0347
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
|