mirror of
https://github.com/0xSojalSec/airllm.git
synced 2026-03-08 06:43:15 +00:00
31 lines
914 B
Python
31 lines
914 B
Python
import setuptools
|
|
|
|
with open("README.md", "r") as fh:
|
|
long_description = fh.read()
|
|
|
|
setuptools.setup(
|
|
name="airllm",
|
|
version="0.9.5",
|
|
author="Gavin Li",
|
|
author_email="gavinli@animaai.cloud",
|
|
description="AirLLM allows single 4GB GPU card to run 70B large language models without quantization, distillation or pruning.",
|
|
long_description=long_description,
|
|
long_description_content_type="text/markdown",
|
|
url="https://github.com/lyogavin/Anima/tree/main/air_llm",
|
|
packages=setuptools.find_packages(),
|
|
install_requires=[ # I get to this in a second
|
|
'tqdm',
|
|
'torch',
|
|
'transformers',
|
|
'accelerate',
|
|
'safetensors',
|
|
'optimum',
|
|
'huggingface_hub'
|
|
],
|
|
classifiers=[
|
|
"Programming Language :: Python :: 3",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Operating System :: OS Independent",
|
|
],
|
|
)
|