diff --git a/auto_round/export/export_to_itrex/bit_packer.py b/auto_round/export/export_to_itrex/bit_packer.py index 44cd708ff..d3f323a6c 100644 --- a/auto_round/export/export_to_itrex/bit_packer.py +++ b/auto_round/export/export_to_itrex/bit_packer.py @@ -16,7 +16,10 @@ from typing import Callable, Dict, Tuple -import numba +try: + import numba +except ImportError: + raise ImportError("Numba is required to use bit_packer module. Please install numba with `pip install numba`.") import numpy as np # key: (bits, compress_bits), value: pack function diff --git a/requirements.txt b/requirements.txt index 6b432993d..c142a76b1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,6 @@ numpy < 2.0 tqdm packaging pillow -numba tbb torch transformers>=4.38 diff --git a/setup.py b/setup.py index 338683125..152359b19 100644 --- a/setup.py +++ b/setup.py @@ -154,7 +154,7 @@ def fetch_requirements(path): packages=include_packages, install_requires=install_requires, extras_require=extras_require, - python_requires=">=3.7.0", + python_requires=">=3.10.0", classifiers=[ "Intended Audience :: Science/Research", "Programming Language :: Python :: 3", diff --git a/test/test_cpu/requirements.txt b/test/test_cpu/requirements.txt index 9ce86165d..976d2e2c3 100644 --- a/test/test_cpu/requirements.txt +++ b/test/test_cpu/requirements.txt @@ -4,3 +4,4 @@ gguf torchvision compressed-tensors parameterized +numba \ No newline at end of file diff --git a/test/test_cuda/requirements.txt b/test/test_cuda/requirements.txt index a1f11760c..37b596c47 100644 --- a/test/test_cuda/requirements.txt +++ b/test/test_cuda/requirements.txt @@ -16,6 +16,7 @@ py-cpuinfo torch torchvision tqdm +numba transformers vllm>=0.8.5.post1 compressed-tensors