소스 검색

Better handle spaces for Continue

oobabooga 2 년 전
부모
커밋
120f5662cf
1개의 변경된 파일3개의 추가작업 그리고 2개의 파일을 삭제
  1. 3 2
      modules/chat.py

+ 3 - 2
modules/chat.py

@@ -171,8 +171,9 @@ def chatbot_wrapper(text, generate_state, name1, name2, context, mode, end_of_tu
                     shared.history['visible'].append(['', ''])
 
             if _continue:
-                shared.history['internal'][-1] = [text, f'{last_reply[0]} {reply}']
-                shared.history['visible'][-1] = [visible_text, f'{last_reply[1]} {visible_reply}']
+                sep = list(map(lambda x : ' ' if x[-1] != ' ' else '', last_reply))
+                shared.history['internal'][-1] = [text, f'{last_reply[0]}{sep[0]}{reply}']
+                shared.history['visible'][-1] = [visible_text, f'{last_reply[1]}{sep[1]}{visible_reply}']
             else:
                 shared.history['internal'][-1] = [text, reply]
                 shared.history['visible'][-1] = [visible_text, visible_reply]