File size: 1,448 Bytes
176b11a 45c83c9 176b11a | 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 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | [build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "sparsevlm"
version = "0.1.3"
description = "Training-free visual token sparsification for vision-language models (ICML 2025)"
readme = "README.md"
license = { text = "Apache-2.0" }
authors = [{ name = "Aryan Chauhan", email = "chauhanaryan31801@gmail.com" }]
keywords = ["vision-language-models", "token-pruning", "inference-optimization", "transformers"]
classifiers = [
"Development Status :: 3 - Alpha",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: Apache Software License",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
]
requires-python = ">=3.10"
dependencies = [
"torch>=2.1.0",
"transformers>=4.40.0",
"numpy>=1.24.0",
]
[project.optional-dependencies]
triton = ["triton>=2.1.0"]
dev = ["pytest>=7.0", "pytest-cov", "Pillow", "accelerate"]
[project.urls]
Homepage = "https://github.com/aryanchauhan31/SparseVLM"
Repository = "https://github.com/aryanchauhan31/SparseVLM"
Paper = "https://arxiv.org/abs/2410.04417"
[tool.setuptools.packages.find]
where = ["."]
include = ["sparsevlm*", "kernels*"]
[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = ["test_*.py"]
addopts = "-v --tb=short"
|