pyproject.toml 739 Bytes
Newer Older
Ji Lin's avatar
Ji Lin committed
1
2
3
4
5
6
7
8
9
[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[project]
name = "awq"
version = "0.1.0"
description = "An efficient and accurate low-bit weight quantization(INT3/4) method for LLMs."
readme = "README.md"
Haotian Tang's avatar
Haotian Tang committed
10
requires-python = ">=3.8"
Ji Lin's avatar
Ji Lin committed
11
12
13
14
15
16
17
classifiers = [
    "Programming Language :: Python :: 3",
    "License :: OSI Approved :: Apache Software License",
]
dependencies = [
    "accelerate", "sentencepiece", "tokenizers>=0.12.1",
    "torch", "torchvision", 
Haotian Tang's avatar
Haotian Tang committed
18
19
20
21
    "transformers>=4.31.0", 
    "lm_eval", "texttable",
    "toml", "attributedict",
    "protobuf"
Ji Lin's avatar
Ji Lin committed
22
23
24
25
26
27
]

[tool.setuptools.packages.find]
exclude = ["results*", "scripts*", "examples*"]

[tool.wheel]
Haotian Tang's avatar
Haotian Tang committed
28
exclude = ["results*", "scripts*", "examples*"]