Wheelodex » Projects » tuned-lens

tuned-lens

View on PyPI — Reverse Dependencies (0)

0.2.0 tuned_lens-0.2.0-py3-none-any.whl

Wheel Details

Project: tuned-lens
Version: 0.2.0
Filename: tuned_lens-0.2.0-py3-none-any.whl
Download: [link]
Size: 56865
MD5: 062b2d235153cf6973413d2936ec515d
SHA256: 1e41b8dad117984047e806529ceca620869d630aecead949ec5ca53d09b4f6ee
Uploaded: 2023-07-18 17:04:39 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt · entry_points.txt

METADATA

Metadata-Version: 2.1
Name: tuned-lens
Version: 0.2.0
Summary: Tools for understanding how transformer predictions are built layer-by-layer
License: MIT License
Keywords: nlp,interpretability,language-models,explainable-ai
Requires-Python: >=3.9
Requires-Dist: accelerate
Requires-Dist: datasets
Requires-Dist: plotly (>=5.13.1)
Requires-Dist: torchdata (>=0.6.0)
Requires-Dist: torch (>=1.13.0)
Requires-Dist: transformers (>=4.28.1)
Requires-Dist: huggingface-hub (>=0.12.1)
Requires-Dist: simple-parsing (>=0.1.1)
Requires-Dist: flatten-dict (>=0.4.1)
Requires-Dist: wandb (>=0.15.0)
Requires-Dist: bitsandbytes; extra == "8bit"
Requires-Dist: pre-commit; extra == "dev"
Requires-Dist: bump2version; extra == "dev"
Requires-Dist: pytest-skip-slow; extra == "dev"
Requires-Dist: pytest; extra == "dev"
Requires-Dist: furo; extra == "docs"
Requires-Dist: myst-parser; extra == "docs"
Requires-Dist: sphinx; extra == "docs"
Requires-Dist: sphinx-autodoc-typehints; extra == "docs"
Requires-Dist: sphinx-rtd-theme; extra == "docs"
Requires-Dist: nbsphinx; extra == "docs"
Requires-Dist: ipywidgets; extra == "notebooks"
Requires-Dist: sentencepiece (>=0.1.99); extra == "slow_tokenizers"
Requires-Dist: sentencepiece (>=0.1.99); extra == "test"
Requires-Dist: transformer-lens (>=1.2.2); extra == "test"
Requires-Dist: pytest-skip-slow; extra == "test"
Requires-Dist: pytest-cov; extra == "test"
Requires-Dist: nbmake; extra == "test"
Requires-Dist: pytest; extra == "test"
Requires-Dist: mock; extra == "test"
Requires-Dist: transformer-lens (>=1.2.2); extra == "transformer_lens"
Provides-Extra: 8bit
Provides-Extra: dev
Provides-Extra: docs
Provides-Extra: notebooks
Provides-Extra: slow_tokenizers
Provides-Extra: test
Provides-Extra: transformer_lens
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 3771 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.40.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
tests/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/conftest.py sha256=eeoxJ8A4gwNCccfoLamhXv8xWfic4DVsxgnpweuI4SY 2608
tests/test_data.py sha256=FfUjdjdCVFT-X6QRHI59y945QVVH5Jrx1ehJwBI6VRA 1438
tests/test_distance.py sha256=vzXejCNBxrfpGNHDVmCD4n4QoAvy-OydLmWHOmUxOPw 940
tests/test_lenses.py sha256=tl1qur69KTbooF4tBW6oLDQW8je3XgdvnCUiqfOz9aE 3842
tests/test_load_artifact.py sha256=AAoxf4psdQcEEULqRs075aKtkhwnvNGTzVJwZVA28L4 161
tests/test_model_surgery.py sha256=c8mkYJM_GIiZ4-hMsneTVJPdikRCRNgXGpqBMo2s7n0 870
tests/test_stats.py sha256=Gn41y0D5NfCc8DH7m9Oj0mTFpX2cS4CvQOdoow9aYc4 612
tests/test_subspaces.py sha256=G4JhEIywylXyW77THS_X6YPM6zJ6sbM5rWBRYEvn56o 609
tests/test_unembed.py sha256=6t7LEPzUeyoC_6Tx6cZcIRNiq7NTJdO79MH16cQYiSg 1201
tests/test_utils.py sha256=OVsYGSVDBSZunBLyEvVtUV4O3hjCGMFOsBpJORoEzys 393
tests/scripts/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/scripts/test_integration.py sha256=KUDDiwK9mC3LAnJt9iVD7AIM0EQvhalejZkKWpZyC98 1122
tuned_lens/__init__.py sha256=1mDVOFGo92RHD6Jkx6blD8yWiszkzEoWnRERXZBUQ08 56
tuned_lens/__main__.py sha256=SPwFYbYcWUVfGwCl1OvcBLn1rsnOMNlPlu9u655kg3Q 869
tuned_lens/data.py sha256=cRZQHPC16ClahoTVQ9pgoFTWwEiXeJv1dm8bcnNEicQ 5134
tuned_lens/load_artifacts.py sha256=UjS2vfVZqtMpU4j3n8n_EZr29XZ3s44CJES1Hf5fkLw 2895
tuned_lens/model_surgery.py sha256=v_bNZhN-F9k1LmCemGdz-hERItOj5ZmS79P5RCUXfoE 7768
tuned_lens/utils.py sha256=N05q_V6Gm259SI79uB3a_XzaSJfrs1L9pcLFZAFfts4 7974
tuned_lens/causal/__init__.py sha256=a3Xfpe_JfWyBvaKK4YQnSobHOPooTPPS_cnfwV6WWKY 246
tuned_lens/causal/ablation.py sha256=9sA0G8cYMc5CimN9Xpn9J-zR1rVuc3UYcnqRTdUSsQk 2066
tuned_lens/causal/subspaces.py sha256=2GXzIY0Ia8iYyVOJfg_NB3mCtvXia7aSynHDfZ3jeJM 9031
tuned_lens/causal/utils.py sha256=AkNSeNLq3ro6_GLDJgS3SrDgaMYgO4FfNhXapKxg5_o 1311
tuned_lens/nn/__init__.py sha256=89BV--Tt6QuYEkoYqxYBbUn7wh0jzXanIC-khR7CbKI 203
tuned_lens/nn/lenses.py sha256=bwaCmQbGE32eGW5S1z5EK_B0iU2wfcYL9Tr6G7ptYlQ 10175
tuned_lens/nn/unembed.py sha256=xcXNvqsyr8c5EgowVphz82e92jSliF5gzBV7f2SpoF0 6790
tuned_lens/plotting/__init__.py sha256=yXytF8mUoJ2ptJkrqx6wftlyrxSXxBSW3ImD9x6uNHQ 206
tuned_lens/plotting/prediction_trajectory.py sha256=8s1NVU2q39g36-YWMbsP-1O9wnyFUZzb3lccLrWVM_Q 28184
tuned_lens/plotting/token_formatter.py sha256=5249nqj3lNo_EqA_wMhuEzC5LhEFR6gsQH6efmb6n3w 1424
tuned_lens/plotting/trajectory_plotting.py sha256=B7wjIXIfzdvElLn1Hz_bye6b3nWTQRfSsPQCsIY3Wt8 9454
tuned_lens/scripts/__init__.py sha256=aEHg3yBVhWZQA9Hro4fMqnR4gi6gMVHjHJge2n9cG54 38
tuned_lens/scripts/eval_loop.py sha256=2UhmXrxeNOwqt-Od5ES5skT9RIzZp94TVy-bEG7v78I 12070
tuned_lens/scripts/ingredients.py sha256=qtvcJuCtVEcbtXdrrG_tcd30ihHRwo01kV89EKBoFH8 13264
tuned_lens/scripts/train_loop.py sha256=BH78dyrgwt4x_13KJgdZywapnf_6oAI20GueX8LwDAU 16609
tuned_lens/stats/__init__.py sha256=AQZ-pH35KI5KG3zCctm3xFHet19G4bpCO889FGNB2XQ 167
tuned_lens/stats/distance.py sha256=YSvSLLt54fn9J4nk2Dvdovzbu3mdP6pLoivxEdDbG7o 1857
tuned_lens/stats/logit_stats.py sha256=cIl4tKYGC0BE9TquGpaiqnO1cVY14Yc8D888n6x2tKA 4121
tuned_lens-0.2.0.dist-info/LICENSE sha256=dlfiknmzqG6eUKMClGWxqsm99BJmBkB51lhaUHtzt-c 1069
tuned_lens-0.2.0.dist-info/METADATA sha256=XYk2iNTYnE1NxHOLAOo1zN_mosy6lHrSORVmQnU6ToY 5613
tuned_lens-0.2.0.dist-info/WHEEL sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18 92
tuned_lens-0.2.0.dist-info/entry_points.txt sha256=lW0KREPtaZGIeu86l2adDY6ZwQj4ojYLaKgw9YCjY8U 56
tuned_lens-0.2.0.dist-info/top_level.txt sha256=Q3u3HT1wdUDCvHryLavjyGmkNhpCur_Q_BC056gUTrI 17
tuned_lens-0.2.0.dist-info/RECORD — —

top_level.txt

tests
tuned_lens

entry_points.txt

[console_scripts]
tuned-lens = tuned_lens.__main__:main