Skip to content
Snippets Groups Projects
Commit 8b98e698 authored by Dipam Chakraborty's avatar Dipam Chakraborty
Browse files

fix: no need to remove last dialogue

parent 03955d20
No related branches found
No related tags found
No related merge requests found
......@@ -50,7 +50,7 @@ class BARTResponseAgent(object):
persona = [tokenize(line.strip()) for line in conversation['persona B']]
# partner = [tokenize(line.strip()) for line in conversation['persona A']]
partner = [] # Baseline not trained with the partner personaj
history = [tokenize(line['text'].strip()) for line in conversation['dialogue'][:-1]]
history = [tokenize(line['text'].strip()) for line in conversation['dialogue']]
return persona, partner, history
def prepare_tensors(self, conversation):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment