KKwargs don't propogate to `load_lens_artifacts` and `th.load`

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

KKwargs don't propogate to `load_lens_artifacts` and `th.load` #85

Closed levmckinney closed 1 year ago

levmckinney commented 1 year ago

When calling TunedLens.from_model_and_pretrained arguments like revision and map_location and not propagated to load_lens_artifacts and th.load respectively. Leading to an unknown kwargs error.