| from setuptools import setup, find_packages |
|
|
| setup( |
| name="unimodal-sith", |
| version="0.1.0", |
| description="UniSITH: Unimodal Semantic Inspection of Transformer Heads", |
| long_description=open("README.md").read(), |
| long_description_content_type="text/markdown", |
| author="Adapted from SITH (Vaquero et al., 2025)", |
| url="https://huggingface.co/lorenzovaquero/UniSITH", |
| packages=find_packages(), |
| python_requires=">=3.9", |
| install_requires=[ |
| "torch>=2.0", |
| "transformers>=4.30", |
| "datasets>=2.0", |
| "scipy>=1.10", |
| "Pillow>=9.0", |
| "tqdm>=4.60", |
| "numpy>=1.24", |
| ], |
| extras_require={ |
| "clip": ["open-clip-torch>=2.20"], |
| }, |
| classifiers=[ |
| "Development Status :: 3 - Alpha", |
| "Intended Audience :: Science/Research", |
| "Topic :: Scientific/Engineering :: Artificial Intelligence", |
| "Programming Language :: Python :: 3", |
| ], |
| ) |
|
|
