Adding GENRE model to ecco

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

Adding GENRE model to ecco #94

Open sky-2002 opened 1 year ago

sky-2002 commented 1 year ago

Hi @jalammar , thanks a lot for this amazing package and the amazing blogs you have created. I am using the facebook GENRE model for entity linking. I wanted to use ecco with it. How to add it to ecco? I tried changing model_config, but not sure how to do it exactly, I also followed the notebook - Identifying model configuration.ipynb, but ran into errors. Can you provide some guidance on how to add new models? Would be of great help.

BiEchi commented 1 year ago

What error did you encounter exactly? I'm also trying to add a new model