huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 12:24:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revert "Use optimum-habana v1.15-release branch"

Browse Source 
This reverts commit c6f023a06b.
This commit is contained in:
yuanwu 2024-12-11 08:17:17 +00:00
parent c922ef9534
commit c3b8899f10
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/pyproject.toml
View File

@ -22,7 +22,7 @@ opentelemetry-instrumentation-grpc = "^0.36b0"
hf-transfer = "^0.1.2"
sentencepiece = "^0.1.97"
peft = "^0.10"
#optimum-habana = "1.15.0"
optimum-habana = "1.14.1"
transformers = "4.45.2"
numpy = "1.26.4"
accelerate = "0.33.0"

3
server/requirements.txt
View File

@ -46,8 +46,7 @@ opentelemetry-instrumentation==0.36b0 ; python_version >= "3.9" and python_versi
opentelemetry-proto==1.15.0 ; python_version >= "3.9" and python_version < "3.13"
opentelemetry-sdk==1.15.0 ; python_version >= "3.9" and python_version < "3.13"
opentelemetry-semantic-conventions==0.36b0 ; python_version >= "3.9" and python_version < "3.13"
#optimum-habana==1.14.1 ; python_version >= "3.9" and python_version < "3.13"
optimum-habana @ git+https://github.com/huggingface/optimum-habana.git@v1.15-release ; python_version >= "3.9" and python_version < "3.13"
optimum-habana==1.14.1 ; python_version >= "3.9" and python_version < "3.13"
optimum==1.23.2 ; python_version >= "3.9" and python_version < "3.13"
packaging==24.1 ; python_version >= "3.9" and python_version < "3.13"
pandas==2.2.3 ; python_version >= "3.9" and python_version < "3.13"