Update optimum-habana version to 1.11.1 (#138) (#142)

Co-authored-by: Karol Damaszke <kdamaszke@habana.ai>
This commit is contained in:
Karol Damaszke 2024-05-14 14:13:31 +02:00 committed by GitHub
parent 91b0aa69d1
commit 6afd82a1e7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 13 additions and 7 deletions

11
server/poetry.lock generated
View File

@ -1800,25 +1800,26 @@ tests = ["Pillow", "accelerate", "diffusers (>=0.17.0)", "einops", "invisible-wa
[[package]]
name = "optimum-habana"
version = "1.11.0"
version = "1.11.1"
description = "Optimum Habana is the interface between the Hugging Face Transformers and Diffusers libraries and Habana's Gaudi processor (HPU). It provides a set of tools enabling easy model loading, training and inference on single- and multi-HPU settings for different downstream tasks."
optional = false
python-versions = "*"
files = [
{file = "optimum-habana-1.11.0.tar.gz", hash = "sha256:d7c7caea5daf17c0d520c52797d8a663432de6aa9ee8b35f6304d1646f225826"},
{file = "optimum_habana-1.11.0-py3-none-any.whl", hash = "sha256:7e94b40d339ac42f754d6e0c9f7df513ebd1afd63307ec2f05e677adfbe347c4"},
{file = "optimum_habana-1.11.1-py3-none-any.whl", hash = "sha256:2b4f2a801ebd7f235bde91f3e351893efcc586afef3b0c867eb3c4c8c4c26bec"},
{file = "optimum_habana-1.11.1.tar.gz", hash = "sha256:c70dbd6d69ce11c8bb78292913304616600d051d94237c8310c3bdbf73ca7017"},
]
[package.dependencies]
accelerate = "<0.28.0"
diffusers = ">=0.26.0,<0.27.0"
optimum = "*"
pytest = "<8.0.0"
torch = "*"
transformers = ">=4.38.0,<4.39.0"
[package.extras]
quality = ["hf-doc-builder", "ruff"]
tests = ["GitPython", "datasets", "optuna", "parameterized", "psutil", "pytest (<8.0.0)", "safetensors", "sentencepiece"]
tests = ["GitPython", "datasets", "optuna", "parameterized", "psutil", "safetensors", "sentencepiece"]
[[package]]
name = "outlines"
@ -3612,4 +3613,4 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "more-itertools", "p
[metadata]
lock-version = "2.0"
python-versions = ">=3.9,<3.13"
content-hash = "d7b7c690db90064982f33b8855d22434ea5496b4da886e9df9d28fcc3f990f3b"
content-hash = "f54475fbf8d56bf0ff0b26fbc12afb160648dd03aaefbd8ae636b920964d39f0"

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.11.0"
optimum-habana = "1.11.1"
transformers = "4.38.2"
accelerate = "0.27.2"
outlines= { version = "^0.0.36", optional = true }

View File

@ -13,6 +13,7 @@ datasets==2.19.1 ; python_version >= "3.9" and python_version < "3.13"
deprecated==1.2.14 ; python_version >= "3.9" and python_version < "3.13"
diffusers==0.26.3 ; python_version >= "3.9" and python_version < "3.13"
dill==0.3.8 ; python_version >= "3.9" and python_version < "3.13"
exceptiongroup==1.2.1 ; python_version >= "3.9" and python_version < "3.11"
filelock==3.14.0 ; python_version >= "3.9" and python_version < "3.13"
frozenlist==1.4.1 ; python_version >= "3.9" and python_version < "3.13"
fsspec==2024.3.1 ; python_version >= "3.9" and python_version < "3.13"
@ -27,6 +28,7 @@ huggingface-hub==0.23.0 ; python_version >= "3.9" and python_version < "3.13"
humanfriendly==10.0 ; python_version >= "3.9" and python_version < "3.13"
idna==3.7 ; python_version >= "3.9" and python_version < "3.13"
importlib-metadata==7.1.0 ; python_version >= "3.9" and python_version < "3.13"
iniconfig==2.0.0 ; python_version >= "3.9" and python_version < "3.13"
intel-openmp==2021.4.0 ; python_version >= "3.9" and python_version < "3.13" and platform_system == "Windows"
jinja2==3.1.4 ; python_version >= "3.9" and python_version < "3.13"
loguru==0.6.0 ; python_version >= "3.9" and python_version < "3.13"
@ -46,17 +48,19 @@ 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.11.0 ; python_version >= "3.9" and python_version < "3.13"
optimum-habana==1.11.1 ; python_version >= "3.9" and python_version < "3.13"
optimum==1.19.2 ; python_version >= "3.9" and python_version < "3.13"
packaging==24.0 ; python_version >= "3.9" and python_version < "3.13"
pandas==2.2.2 ; python_version >= "3.9" and python_version < "3.13"
peft==0.10.0 ; python_version >= "3.9" and python_version < "3.13"
pillow==10.3.0 ; python_version >= "3.9" and python_version < "3.13"
pluggy==1.5.0 ; python_version >= "3.9" and python_version < "3.13"
protobuf==3.20.3 ; python_version >= "3.9" and python_version < "3.13"
psutil==5.9.8 ; python_version >= "3.9" and python_version < "3.13"
pyarrow-hotfix==0.6 ; python_version >= "3.9" and python_version < "3.13"
pyarrow==16.0.0 ; python_version >= "3.9" and python_version < "3.13"
pyreadline3==3.4.1 ; sys_platform == "win32" and python_version >= "3.9" and python_version < "3.13"
pytest==7.4.4 ; python_version >= "3.9" and python_version < "3.13"
python-dateutil==2.9.0.post0 ; python_version >= "3.9" and python_version < "3.13"
pytz==2024.1 ; python_version >= "3.9" and python_version < "3.13"
pyyaml==6.0.1 ; python_version >= "3.9" and python_version < "3.13"
@ -69,6 +73,7 @@ six==1.16.0 ; python_version >= "3.9" and python_version < "3.13"
sympy==1.12 ; python_version >= "3.9" and python_version < "3.13"
tbb==2021.12.0 ; python_version >= "3.9" and python_version < "3.13" and platform_system == "Windows"
tokenizers==0.15.2 ; python_version >= "3.9" and python_version < "3.13"
tomli==2.0.1 ; python_version >= "3.9" and python_version < "3.11"
tqdm==4.66.4 ; python_version >= "3.9" and python_version < "3.13"
transformers==4.38.2 ; python_version >= "3.9" and python_version < "3.13"
transformers[sentencepiece]==4.38.2 ; python_version >= "3.9" and python_version < "3.13"