diff --git a/backend/python/transformers/requirements-cpu.txt b/backend/python/transformers/requirements-cpu.txt index 421c4b80..c88508e3 100644 --- a/backend/python/transformers/requirements-cpu.txt +++ b/backend/python/transformers/requirements-cpu.txt @@ -1,7 +1,8 @@ torch==2.4.1 llvmlite==0.43.0 +numba==0.60.0 accelerate transformers bitsandbytes outetts -sentence-transformers==3.3.1 +sentence-transformers==3.3.1 \ No newline at end of file diff --git a/backend/python/transformers/requirements-cublas11.txt b/backend/python/transformers/requirements-cublas11.txt index c5d18d09..0faa9cec 100644 --- a/backend/python/transformers/requirements-cublas11.txt +++ b/backend/python/transformers/requirements-cublas11.txt @@ -1,6 +1,7 @@ --extra-index-url https://download.pytorch.org/whl/cu118 torch==2.4.1+cu118 llvmlite==0.43.0 +numba==0.60.0 accelerate transformers bitsandbytes diff --git a/backend/python/transformers/requirements-cublas12.txt b/backend/python/transformers/requirements-cublas12.txt index c0bcfc87..1e22312f 100644 --- a/backend/python/transformers/requirements-cublas12.txt +++ b/backend/python/transformers/requirements-cublas12.txt @@ -1,6 +1,7 @@ torch==2.4.1 accelerate llvmlite==0.43.0 +numba==0.60.0 transformers bitsandbytes outetts diff --git a/backend/python/transformers/requirements-hipblas.txt b/backend/python/transformers/requirements-hipblas.txt index e7f53860..47aa88db 100644 --- a/backend/python/transformers/requirements-hipblas.txt +++ b/backend/python/transformers/requirements-hipblas.txt @@ -3,6 +3,7 @@ torch==2.4.1+rocm6.0 accelerate transformers llvmlite==0.43.0 +numba==0.60.0 bitsandbytes outetts bitsandbytes diff --git a/backend/python/transformers/requirements-intel.txt b/backend/python/transformers/requirements-intel.txt index aada6e00..708b0516 100644 --- a/backend/python/transformers/requirements-intel.txt +++ b/backend/python/transformers/requirements-intel.txt @@ -4,6 +4,7 @@ torch==2.3.1+cxx11.abi oneccl_bind_pt==2.3.100+xpu optimum[openvino] llvmlite==0.43.0 +numba==0.60.0 intel-extension-for-transformers bitsandbytes outetts diff --git a/backend/python/transformers/requirements.txt b/backend/python/transformers/requirements.txt index d353e4d0..db41b928 100644 --- a/backend/python/transformers/requirements.txt +++ b/backend/python/transformers/requirements.txt @@ -3,5 +3,4 @@ protobuf certifi setuptools scipy==1.15.1 -numpy>=2.0.0 -numba==0.60.0 \ No newline at end of file +numpy>=2.0.0 \ No newline at end of file