Forráskód Böngészése

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

Maya 2 éve
szülő
commit
8fa182cfa7
1 módosított fájl, 1 hozzáadás és 1 törlés
  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):
-    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)
     else:
         last_visible = shared.history['visible'].pop()