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

Fix kwarg propagation in `TunedLens.from_model_and_pretrained()` #84

Closed levmckinney closed 1 year ago

levmckinney commented 1 year ago

Fixes #85.

codecov[bot] commented 1 year ago

Codecov Report

Merging #84 (6260ecc) into main (f396790) will increase coverage by 3.43%. The diff coverage is 100.00%.

Impacted file tree graph

@@            Coverage Diff             @@
##             main      #84      +/-   ##
==========================================
+ Coverage   75.18%   78.62%   +3.43%     
==========================================
  Files          32       32              
  Lines        1769     1740      -29     
==========================================
+ Hits         1330     1368      +38     
+ Misses        439      372      -67     
Impacted Files Coverage Δ
tests/test_lenses.py 86.56% <100.00%> (+1.56%) :arrow_up:
tuned_lens/nn/lenses.py 91.66% <100.00%> (-0.16%) :arrow_down:

... and 5 files with indirect coverage changes