I get this message for the test you provide
#2
by
JeisonJA
- opened
TypeError: argument 'ids': 'list' object cannot be interpreted as an integer
I encountered this error as well, and the following modification worked for me. You may try to change the
'''output_text = tokenizer.decode(outputs, skip_special_tokens=True)'''
to
'''output_text = tokenizer.batch_decode(outputs, skip_special_tokens=True)'''