t_generated = [ ]
for j in range(nb_tokens):
- t = [ [ self.token2id[u] for u in t_primer + t_generated ] ]
+ t = [ [ self.token2id[u] for u in t_primer + t_generated ] + [ 0 ] ]
input = torch.tensor(t, device = self.device)
output = model(input)
logits = output[0, -1]