Spaces:
Runtime error
Runtime error
fix rp too
Browse files
app.py
CHANGED
@@ -359,9 +359,10 @@ def open_rp_chat(model_name, history, system_msg, max_new_tokens, temperature, t
|
|
359 |
|
360 |
model_res = model(messages, config=config) # type: Generator[List[Dict[str, str]], None, None]
|
361 |
for res in model_res:
|
362 |
-
tokens = re.findall(r'\
|
363 |
# tokens = re.findall(r'\s*\S+\s*', res[0]['generated_text'])
|
364 |
for subtoken in tokens:
|
|
|
365 |
history[-1][1] += subtoken
|
366 |
# stream the response
|
367 |
yield history, history, ""
|
|
|
359 |
|
360 |
model_res = model(messages, config=config) # type: Generator[List[Dict[str, str]], None, None]
|
361 |
for res in model_res:
|
362 |
+
tokens = re.findall(r'(.*?)(\s|$)', res[0]['generated_text'])
|
363 |
# tokens = re.findall(r'\s*\S+\s*', res[0]['generated_text'])
|
364 |
for subtoken in tokens:
|
365 |
+
subtoken = "".join(subtoken)
|
366 |
history[-1][1] += subtoken
|
367 |
# stream the response
|
368 |
yield history, history, ""
|