Fixes issue with running the interactive notebook on a cuda device - Githubissues

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

Fixes issue with running the interactive notebook on a cuda device #54

Closed levmckinney closed 1 year ago

levmckinney commented 1 year ago

This should resolve #53