File size: 508 Bytes
69202a6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | 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",
],
)
|