Просмотр исходного кода

Print the tokenized example dialogue in a prettier way

oobabooga 3 лет назад
Родитель
Сommit
2dadf42cb5
1 измененных файлов с 8 добавлено и 4 удалено
  1. 8 4
      server.py

+ 8 - 4
server.py

@@ -519,10 +519,14 @@ if args.chat or args.cai_chat:
                     _history.append(entry)
                     _history.append(entry)
                 entry = ['', '']
                 entry = ['', '']
 
 
-        print(f"\nDialogue tokenized to (formatted as [question, reply]):\n\n", end='')
-        for i in _history:
-            print(i)
-        print("--------------------\n", end='')
+        print(f"\033[1;32;1m\nDialogue tokenized to:\033[0;37;0m\n", end='')
+        for row in _history:
+            for column in row:
+                print("\n")
+                for line in column.strip().split('\n'):
+                    print("|  "+line+"\n")
+                print("|\n")
+            print("------------------------------")
 
 
         return _history
         return _history