xu song
commited on
Commit
•
1b2b08c
1
Parent(s):
04a7868
update
Browse files- models/cpp_qwen2.py +1 -1
models/cpp_qwen2.py
CHANGED
@@ -69,7 +69,7 @@ class Qwen2Simulator(Simulator):
|
|
69 |
"<|im_start|>",
|
70 |
"<|endoftext|>",
|
71 |
]
|
72 |
-
self.stop_tokens = self.tokenize("".
|
73 |
self.generation_kwargs = dict(
|
74 |
temperature=config.DEFAULT_TEMPERATURE,
|
75 |
top_p=config.DEFAULT_TOP_P,
|
|
|
69 |
"<|im_start|>",
|
70 |
"<|endoftext|>",
|
71 |
]
|
72 |
+
self.stop_tokens = self.tokenize("".join(self.stop_words))
|
73 |
self.generation_kwargs = dict(
|
74 |
temperature=config.DEFAULT_TEMPERATURE,
|
75 |
top_p=config.DEFAULT_TOP_P,
|