nikravan commited on
Commit
a259ca0
Β·
verified Β·
1 Parent(s): 1700be6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -87,8 +87,8 @@ def predict(message, history, system_prompt, temperature, max_new_tokens, top_k,
87
  outputs = []
88
  for new_token in streamer:
89
  outputs.append(new_token)
90
- if new_token in stop_tokens:
91
- break
92
  yield "".join(outputs)
93
 
94
 
 
87
  outputs = []
88
  for new_token in streamer:
89
  outputs.append(new_token)
90
+ #if new_token in stop_tokens:
91
+ #break
92
  yield "".join(outputs)
93
 
94