Closed alyonavyshnevska closed 3 years ago
I just ran into this. This is because we're using an updated version of transformers. The fix is actually really simple. In interact.py in the sample_sequence method change: logits = model(input_ids, token_type_ids=token_type_ids) to logits = model(input_ids, token_type_ids=token_type_ids).logits
@JohnnyC08 thank you, that was it! Now the script works for me
I followed the steps to install all the requirements, and ran the
interact.py
script to download the pre-trained model from S3.below is the error message:
python version: 3.8