| from setuptools import setup, find_packages | |
| setup( | |
| name="cgc-v1", | |
| version="1.0.0", | |
| description="Capability-Guided Compression v1 — GPT-2 Medium experiments", | |
| author="Rishaank Gupta", | |
| packages=find_packages(), | |
| python_requires=">=3.10", | |
| install_requires=[ | |
| "torch>=2.1.0", | |
| "transformers>=4.40.0", | |
| "datasets>=2.18.0", | |
| "scipy>=1.12.0", | |
| "numpy>=1.26.0", | |
| "matplotlib>=3.8.0", | |
| "seaborn>=0.13.0", | |
| "tqdm>=4.66.0", | |
| ], | |
| ) | |