File size: 951 Bytes
7b1435f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | 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",
],
)
|