jalammar / ecco

Explain, analyze, and visualize NLP language models. Ecco creates interactive visualizations directly in Jupyter notebooks explaining the behavior of Transformer-based language models (like GPT2, BERT, RoBERTA, T5, and T0).
https://ecco.readthedocs.io
BSD 3-Clause "New" or "Revised" License
1.96k stars 167 forks source link

Fix error when calling _prepare_decoder_input_ids_for_generation #103

Closed SSamDav closed 1 year ago

SSamDav commented 1 year ago

The code was throwing an error when calling the _prepare_decoder_input_ids_for_generation without the proper parameters. I think this only happened when testing encoder-decoder models. Fix this piece of code by following what huggingface does.