Explorar o código

Fix regeneration of first message in instruct mode (#881)

Maya %!s(int64=2) %!d(string=hai) anos
pai
achega
8fa182cfa7
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      modules/chat.py

+ 1 - 1
modules/chat.py

@@ -206,7 +206,7 @@ def cai_chatbot_wrapper(text, generate_state, name1, name2, context, mode, end_o
 
 
 
 
 def regenerate_wrapper(text, generate_state, name1, name2, context, mode, end_of_turn):
 def regenerate_wrapper(text, generate_state, name1, name2, context, mode, end_of_turn):
-    if (shared.character != 'None' and len(shared.history['visible']) == 1) or len(shared.history['internal']) == 0:
+    if (len(shared.history['visible']) == 1 and not shared.history['visible'][0][0]) or len(shared.history['internal']) == 0:
         yield chat_html_wrapper(shared.history['visible'], name1, name2, mode)
         yield chat_html_wrapper(shared.history['visible'], name1, name2, mode)
     else:
     else:
         last_visible = shared.history['visible'].pop()
         last_visible = shared.history['visible'].pop()