AlignmentResearch / tuned-lens

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

Hub integration #29

Closed levmckinney closed 1 year ago

levmckinney commented 1 year ago

Adds integration with hugging face hub for downloading model. Note that currently you need to be logged in in order to download the models.