فهرست منبع

Improve pygmalion line breaks

oobabooga 3 سال پیش
والد
کامیت
9a7f187b5a
2فایلهای تغییر یافته به همراه2 افزوده شده و 1 حذف شده
  1. 2 1
      server.py
  2. BIN
      webui.png

+ 2 - 1
server.py

@@ -201,10 +201,11 @@ elif args.chat:
             reply = reply[len(question):].split('\n')[0].strip()
         else:
             reply = generate_reply(question, tokens, inference_settings, selected_model)[0]
-            reply = reply[len(question):].strip()
+            reply = reply[len(question):]
             idx = reply.find(f"\n{name1}:")
             if idx != -1:
                 reply = reply[:idx]
+            reply = reply.replace('\n', '\n\n').strip()
 
         history.append((text, reply))
         return history

BIN
webui.png