Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ def generate(
|
|
49 |
conversation.append({"role": "system", "content": system_prompt})
|
50 |
for user, assistant in chat_history:
|
51 |
conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
|
52 |
-
conversation.append({"role": "user", "content": message})
|
53 |
|
54 |
chat = tokenizer.apply_chat_template(conversation, tokenize=False)
|
55 |
inputs = tokenizer(chat, return_tensors="pt", add_special_tokens=False).to("cuda")
|
|
|
49 |
conversation.append({"role": "system", "content": system_prompt})
|
50 |
for user, assistant in chat_history:
|
51 |
conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
|
52 |
+
conversation.append({"role": "user", "content": "### Instruction: \n" + "根据巨杉数据库SequoiaDB的相关问题进行回答。\n" + "### Input:\n" + message + "\n ### Response:"})
|
53 |
|
54 |
chat = tokenizer.apply_chat_template(conversation, tokenize=False)
|
55 |
inputs = tokenizer(chat, return_tensors="pt", add_special_tokens=False).to("cuda")
|