Skip to content

Commit 56f469e

Browse files
authored
remove numba from requirments (#905)
Signed-off-by: yiliu30 <yi4.liu@intel.com>
1 parent 6c8974b commit 56f469e

File tree

5 files changed

+7
-3
lines changed

5 files changed

+7
-3
lines changed

auto_round/export/export_to_itrex/bit_packer.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,10 @@
1616

1717
from typing import Callable, Dict, Tuple
1818

19-
import numba
19+
try:
20+
import numba
21+
except ImportError:
22+
raise ImportError("Numba is required to use bit_packer module. Please install numba with `pip install numba`.")
2023
import numpy as np
2124

2225
# key: (bits, compress_bits), value: pack function

requirements.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@ numpy < 2.0
77
tqdm
88
packaging
99
pillow
10-
numba
1110
tbb
1211
torch
1312
transformers>=4.38

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def fetch_requirements(path):
154154
packages=include_packages,
155155
install_requires=install_requires,
156156
extras_require=extras_require,
157-
python_requires=">=3.7.0",
157+
python_requires=">=3.10.0",
158158
classifiers=[
159159
"Intended Audience :: Science/Research",
160160
"Programming Language :: Python :: 3",

test/test_cpu/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,4 @@ gguf
44
torchvision
55
compressed-tensors
66
parameterized
7+
numba

test/test_cuda/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ py-cpuinfo
1616
torch
1717
torchvision
1818
tqdm
19+
numba
1920
transformers
2021
vllm>=0.8.5.post1
2122
compressed-tensors

0 commit comments

Comments
 (0)