pyvene

View on PyPIReverse Dependencies (1)

0.1.6 pyvene-0.1.6-py3-none-any.whl

Wheel Details

Project: pyvene
Version: 0.1.6
Filename: pyvene-0.1.6-py3-none-any.whl
Download: [link]
Size: 70620
MD5: b39bcd5f0d14b0b6b888136284380008
SHA256: 241453dfeb330af79f42220af8dc61c6287a6f182455fb327848a6ff4a58c4b2
Uploaded: 2024-11-06 01:29:13 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: pyvene
Version: 0.1.6
Summary: Use Activation Intervention to Interpret Causal Mechanism of Model
Author: Zhengxuan Wu
Author-Email: wuzhengx[at]stanford.edu
Home-Page: https://github.com/stanfordnlp/pyvene
License: Apache License 2.0
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: POSIX :: Linux
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Requires-Python: >=3.8
Requires-Dist: torch (>=2.0.0)
Requires-Dist: transformers (>=4.45.1)
Requires-Dist: tokenizers (>=0.20.0)
Requires-Dist: datasets (>=3.0.1)
Requires-Dist: protobuf (>=3.20.0)
Requires-Dist: matplotlib (>=3.7.4)
Requires-Dist: ipywidgets (>=8.1.1)
Requires-Dist: plotnine (>=0.12.4)
Requires-Dist: huggingface-hub (>=0.25.1)
Requires-Dist: numpy (>=1.23.5)
Requires-Dist: fsspec (>=2023.6.0)
Requires-Dist: accelerate (>=0.34.2)
Requires-Dist: sentencepiece (>=0.1.96)
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 3289 characters]

WHEEL

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

RECORD

Path Digest Size
pyvene/__init__.py sha256=NdK-J8vRSYkUYv_GXRgyN3GfIgEGOXY5jrxaT31vBK8 3037
pyvene/data_generators/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/data_generators/causal_model.py sha256=s5mlAMEq48MpmGAZjeFu4MMGzi_RQR2fgh5Cjg8JcJ4 16529
pyvene/models/__init__.py sha256=-6tYOfRuhY6HInqRckeRQ0y6d91wFlvZRtVglgLy-Nc 95
pyvene/models/basic_utils.py sha256=6SzTNz5Ii2L7HHdNQPTgL3LKPsJ2Pn27ft27oh4UrbE 4877
pyvene/models/configuration_intervenable_model.py sha256=bLzca-sAF61QAf6jB8UYT0qPN2DItEFdDfwgwrDwErI 5701
pyvene/models/constants.py sha256=lTRQtOOHH7jCnuQPOAy1fmRoYoRNi6wAoZApN3RkmYU 788
pyvene/models/intervenable_base.py sha256=0mu0xLfO3h77GfHaHaukKbisM0cKSFOlY_Cu9b2q_u4 93831
pyvene/models/intervenable_modelcard.py sha256=AXnjzgK9b8LZAPkB35qNnwgqiv2qVEJZGRc5xszTwX4 6906
pyvene/models/intervention_utils.py sha256=iIeuZPa6G71iyJdUe1-5DjXo_Ck1JRh7aSCW5a6Ha2M 6607
pyvene/models/interventions.py sha256=MBdf57en2auJjeWhSfNZEPsOtOhLl5Ok7drjUSzCzvI 22790
pyvene/models/layers.py sha256=80bkgByV-IiVjG5obwCxYUBjJgMPke8IJ2QAEcV--3A 3283
pyvene/models/modeling_utils.py sha256=k25EfXr7uPl3yTRkVfjs1U8lTVImo5_oswv-8RUFqgQ 19363
pyvene/models/backpack_gpt2/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/models/backpack_gpt2/modelings_backpack_gpt2.py sha256=bMloKHrRU2qDnTNEmZZzQ0sSUfh9bKq8BcQZiYvvKRY 10328
pyvene/models/backpack_gpt2/modelings_intervenable_backpack_gpt2.py sha256=ZgR-45qU26ejRHR86-EpDdJ-2L3GjGHvoZkAFQ0BRHs 1144
pyvene/models/blip/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/models/blip/modelings_blip.py sha256=icUX3KJQMB_bGLjEoIsqKSaAVT1Bgdql5eeRmKWD_O0 3347
pyvene/models/blip/modelings_blip_itm.py sha256=llVDES-oHBQaSvTsLZUwaDEzkIUw59PH9L7e8-cfdOM 3929
pyvene/models/blip/modelings_intervenable_blip.py sha256=6EzAfPJMapBTCATt_SUp2Ol2CZ52JhUNLDkaeKFFyqM 5509
pyvene/models/blip/modelings_intervenable_blip_itm.py sha256=m3etA8OH5hHVB6YdcyDYue_JVsPbmDEbTwnh2SVoO2M 4282
pyvene/models/gemma/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/models/gemma/modelings_intervenable_gemma.py sha256=LdzOy37OcY2mSGDgJSAUUfW67WbS1_bHKARMjKV3Rm0 3494
pyvene/models/gemma2/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/models/gemma2/modelings_intervenable_gemma2.py sha256=aDHIIBX4OcqMp6e0aky4zmqHDH3b2EAL5uvfM3rzwf8 3490
pyvene/models/gpt2/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/models/gpt2/modelings_intervenable_gpt2.py sha256=gmJuBVsziS7jmqKORzq2rs5rnib1Bgm_Ie1RbE7aqjo 4539
pyvene/models/gpt_neo/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/models/gpt_neo/modelings_intervenable_gpt_neo.py sha256=y5OGbTwrTgcSlenEZV9t0qwd6UCokGMxK3zL0FJ7ld8 3005
pyvene/models/gpt_neox/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/models/gpt_neox/modelings_intervenable_gpt_neox.py sha256=YXL5poLzIF0ieHofcxI15AhHGYISY38f_qJV837c1W0 3122
pyvene/models/gru/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/models/gru/modelings_gru.py sha256=gMV1Tgj_ehThjpXMb7YdzFskVwP8SpEBqY4FtIs1Z7U 11328
pyvene/models/gru/modelings_intervenable_gru.py sha256=0Ndu4ejG2RdRVwGW4ls6OciVd5hKwo3L6HZZp1XMLoc 4125
pyvene/models/llama/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/models/llama/modelings_intervenable_llama.py sha256=YVvsWuykSRgJLrI49BXh8kOvOzH23WmGTy4haPwfHbc 3787
pyvene/models/llava/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/models/llava/modelings_intervenable_llava.py sha256=P8wHDPP49Hty3tUorZk5cbL9axGoXk38np664WpBlxg 4256
pyvene/models/mistral/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/models/mistral/modellings_intervenable_mistral.py sha256=mDGLRlthBKOt6au_bt4ZqgcxjSYeD7qfTfDD5KXXV8Y 3315
pyvene/models/mlp/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/models/mlp/modelings_intervenable_mlp.py sha256=FRu17982IEz8LZPcjrwDSlL1PKz8G1QEK9Is1lKDXAE 1502
pyvene/models/mlp/modelings_mlp.py sha256=Um1nweCXHFZUCIuFaTv9UxO-5zANa6ASIlhGagVpzKo 6235
pyvene/models/olmo/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
pyvene/models/olmo/modelings_intervenable_olmo.py sha256=rZ743nJlsFlAeqUsDUN53yxmDF8AQBt8X_mF5Yn_cQI 3816
pyvene-0.1.6.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
pyvene-0.1.6.dist-info/METADATA sha256=1lABFnbNEqZSScA6azH_4lf9g2kk79tiDzCRYUXMqaA 4423
pyvene-0.1.6.dist-info/WHEEL sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc 91
pyvene-0.1.6.dist-info/top_level.txt sha256=s-YyI60UdFkvIWW6i1P8l39-CeYs_Pxgp7Sm5PZysoc 7
pyvene-0.1.6.dist-info/RECORD

top_level.txt

pyvene