| [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" |
|
|
