winglian commited on
Commit
02bf4e7
1 Parent(s): d0d2db4

fix rp too

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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'\S+\s*', res[0]['generated_text'])
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, ""