AlignmentResearch / tuned-lens

Tools for understanding how transformer predictions are built layer-by-layer
https://tuned-lens.readthedocs.io/en/latest/
MIT License
432 stars 47 forks source link

Make it easier to download pretrained lenses #28

Closed levmckinney closed 1 year ago

levmckinney commented 1 year ago

There is currently no easy way to download the lenses.

I a clean solution to solve this would be to integrate this library with hugging face hub.

Another option I've consider is hosting these directly in an s3 bucket but hosting on hugging face should make these models more discoverable and allow us to also host a demo on hugging face spaces.