Print the tokenized example dialogue in a prettier way

This commit is contained in:
oobabooga 2023-01-30 08:29:49 -03:00
parent 161cae001b
commit 2dadf42cb5

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