"Could not find end key, the output is truncated!" message showing during running the model.

#3
by andromeda01111 - opened

I trained the pythia-1b model on a custom dataset. When i am predicting it is showing this message. How can i solve this issue?

can i solve this issue by changing max_length. I tried it but it takes way too much time to do each prediction.

Sign up or log in to comment