search

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

Make dependencies more loose #108

Closed joaonadkarni closed 3 weeks ago

joaonadkarni commented 8 months ago

My main motivation for this change was that I was failing to install this package as it depended on PyYaml 5.4.1 which has some problems - check here. But overall currently it's very hard to install this package because the dependencies are very strict (e.g. no torch 2 etc). They should be loosened up where possible.

I run tox in python 3.11 and the tests locally and with these versions' specs they seem to pass (after a minor fix in the top_k_top_p_filtering_fn). I need to see in the other Python versions with the tests from the GitHub workflows

hadiasghari commented 7 months ago

+1 This PR resolves the problem of installing ecco on more modern install. (Workaround for anyone interested in using it for now: pip install git+https://github.com/joaonadkarni/ecco@update-dependencies)