Tuchuanhuhuhu commited on
Commit
f504a7f
·
1 Parent(s): 8070a3e

没有附加内容时不显示分割线

Browse files
Files changed (1) hide show
  1. modules/models/base_model.py +3 -1
modules/models/base_model.py CHANGED
@@ -225,8 +225,10 @@ class BaseLLMModel:
225
 
226
  stream_iter = self.get_answer_stream_iter()
227
 
 
 
228
  for partial_text in stream_iter:
229
- chatbot[-1] = (chatbot[-1][0], partial_text + "<hr>" +display_append)
230
  self.all_token_counts[-1] += 1
231
  status_text = self.token_message()
232
  yield get_return_value()
 
225
 
226
  stream_iter = self.get_answer_stream_iter()
227
 
228
+ if display_append:
229
+ display_append = "<hr>" +display_append
230
  for partial_text in stream_iter:
231
+ chatbot[-1] = (chatbot[-1][0], partial_text + display_append)
232
  self.all_token_counts[-1] += 1
233
  status_text = self.token_message()
234
  yield get_return_value()