From 260afcd7209e20b0b1bcf5f569adec986494fc8f Mon Sep 17 00:00:00 2001 From: yiliu30 Date: Thu, 16 Oct 2025 06:27:04 -0400 Subject: [PATCH 1/4] remove numba Signed-off-by: yiliu30 --- requirements.txt | 1 - 1 file changed, 1 deletion(-) 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 From bd26c416d3f9a04d70eb48b53f1e366329be5ef6 Mon Sep 17 00:00:00 2001 From: yiliu30 Date: Thu, 16 Oct 2025 06:27:59 -0400 Subject: [PATCH 2/4] add for test Signed-off-by: yiliu30 --- test/test_cpu/requirements.txt | 1 + test/test_cuda/requirements.txt | 1 + 2 files changed, 2 insertions(+) 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 From da9137710726f52210eb7cd1b295793999e57dab Mon Sep 17 00:00:00 2001 From: yiliu30 Date: Thu, 16 Oct 2025 20:51:23 -0400 Subject: [PATCH 3/4] upgrade python version Signed-off-by: yiliu30 --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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", From 6b866f08156a6ca3d89f8aa6809b79bb5a08e2e1 Mon Sep 17 00:00:00 2001 From: yiliu30 Date: Thu, 16 Oct 2025 20:53:06 -0400 Subject: [PATCH 4/4] fix pre-ci Signed-off-by: yiliu30 --- auto_round/export/export_to_itrex/bit_packer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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