| from setuptools import setup, find_packages |
|
|
| setup( |
| name="sensitive-attribute-circuit-analysis", |
| version="0.1.0", |
| description="Mechanistic interpretability study: Do LLMs route through sensitive-attribute circuits that shift analytical choices?", |
| author="ryanyen22", |
| packages=find_packages(), |
| python_requires=">=3.9", |
| install_requires=[ |
| "torch>=2.0.0", |
| "transformers>=4.40.0", |
| "accelerate>=0.27.0", |
| "bitsandbytes>=0.43.0", |
| "scikit-learn>=1.3.0", |
| "numpy>=1.24.0", |
| "matplotlib>=3.7.0", |
| "huggingface_hub>=0.20.0", |
| ], |
| extras_require={ |
| "nnsight": ["nnsight>=0.3.0"], |
| "pyvene": ["pyvene>=0.1.0"], |
| "viz": ["circuitsvis>=1.0.0", "plotly>=5.0.0"], |
| "all": ["nnsight>=0.3.0", "pyvene>=0.1.0", "circuitsvis>=1.0.0"], |
| }, |
| ) |
|
|
