do not return Loss
#118
by
ZZLuan
- opened
I use the code "outputs = self.model(input_ids = input_ids, labels = input_ids)"
however, the "loss = outputs["loss"][0]" is "torch.Size([])"
Llama-3 should return the logits and loss and past_key_values.