Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-12-23 21:18:00 +01:00
Code Issues Packages Projects Releases Wiki Activity

Print the tokenized example dialogue in a prettier way

Browse Source
This commit is contained in:
oobabooga 2023-01-30 08:29:49 -03:00
parent 161cae001b
commit 2dadf42cb5
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
server.py
View File

@ -519,10 +519,14 @@ if args.chat or args.cai_chat:
_history.append(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