yeswondwerr
commited on
Commit
•
6cd45a2
1
Parent(s):
8e9df48
Update __main__.py
Browse files- __main__.py +1 -1
__main__.py
CHANGED
@@ -198,7 +198,7 @@ def answer_question(
|
|
198 |
new_text = (
|
199 |
generated_text
|
200 |
+ "<|start_header_id|>user<|end_header_id|>\n\n"
|
201 |
-
+
|
202 |
+ "<|start_header_id|>assistant<|end_header_id|>\n\n"
|
203 |
)
|
204 |
new_input_ids = tokenizer(new_text, return_tensors="pt").input_ids.to(
|
|
|
198 |
new_text = (
|
199 |
generated_text
|
200 |
+ "<|start_header_id|>user<|end_header_id|>\n\n"
|
201 |
+
+ q
|
202 |
+ "<|start_header_id|>assistant<|end_header_id|>\n\n"
|
203 |
)
|
204 |
new_input_ids = tokenizer(new_text, return_tensors="pt").input_ids.to(
|