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",
    ],
)