1+ from setuptools import setup , find_packages
2+
3+ with open ("README.md" , "r" , encoding = "utf-8" ) as fh :
4+ long_description = fh .read ()
5+
6+ with open ("requirements.txt" , "r" , encoding = "utf-8" ) as fh :
7+ requirements = [line .strip () for line in fh if line .strip () and not line .startswith ("#" )]
8+
9+ setup (
10+ name = "quantllm" ,
11+ version = "0.1.0" ,
12+ author = "Dark Coder" ,
13+ 14+ description = "A lightweight library for quantized LLM fine-tuning and deployment" ,
15+ long_description = long_description ,
16+ long_description_content_type = "text/markdown" ,
17+ url = "https://github.com/codewithdark-git/QuantLLM" ,
18+ packages = find_packages (),
19+ classifiers = [
20+ "Development Status :: 3 - Alpha" ,
21+ "Intended Audience :: Science/Research" ,
22+ "License :: OSI Approved :: MIT License" ,
23+ "Operating System :: OS Independent" ,
24+ "Programming Language :: Python :: 3" ,
25+ "Programming Language :: Python :: 3.8" ,
26+ "Programming Language :: Python :: 3.9" ,
27+ "Topic :: Scientific/Engineering :: Artificial Intelligence" ,
28+ ],
29+ python_requires = ">=3.8" ,
30+ install_requires = requirements ,
31+ extras_require = {
32+ "dev" : [
33+ "sphinx>=4.5.0" ,
34+ "sphinx-rtd-theme>=1.0.0" ,
35+ "sphinx-copybutton>=0.5.0" ,
36+ "sphinx-autodoc-typehints>=1.18.3" ,
37+ "myst-parser>=0.18.1" ,
38+ "pytest>=7.0.0" ,
39+ "pytest-cov>=3.0.0" ,
40+ "black>=22.0.0" ,
41+ "isort>=5.10.0" ,
42+ "flake8>=4.0.0" ,
43+ "torch>=2.0.0" ,
44+ "transformers>=4.30.0" ,
45+ "datasets>=2.12.0" ,
46+ "accelerate>=0.20.0" ,
47+ "peft>=0.4.0" ,
48+ "bitsandbytes>=0.40.0" ,
49+ "scipy>=1.10.0" ,
50+ "scikit-learn>=1.0.0" ,
51+ "tqdm>=4.65.0" ,
52+ "numpy>=1.24.0" ,
53+ "wandb>=0.15.0" ,
54+ "sentencepiece>=0.1.99" ,
55+ "protobuf>=3.20.0" ,
56+ "einops>=0.6.1" ,
57+ "evaluate>=0.4.0" ,
58+ "tensorboard>=2.13.0"
59+ ],
60+ },
61+ include_package_data = True ,
62+ zip_safe = False ,
63+ )
0 commit comments