From 6afd82a1e7fc38d62ac49609618220b96877e1c0 Mon Sep 17 00:00:00 2001 From: Karol Damaszke Date: Tue, 14 May 2024 14:13:31 +0200 Subject: [PATCH] Update optimum-habana version to 1.11.1 (#138) (#142) Co-authored-by: Karol Damaszke --- server/poetry.lock | 11 ++++++----- server/pyproject.toml | 2 +- server/requirements.txt | 7 ++++++- 3 files changed, 13 insertions(+), 7 deletions(-) diff --git a/server/poetry.lock b/server/poetry.lock index b9f3c92d..1d6ecf7a 100644 --- a/server/poetry.lock +++ b/server/poetry.lock @@ -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" diff --git a/server/pyproject.toml b/server/pyproject.toml index 9121797c..f8e57c1e 100644 --- a/server/pyproject.toml +++ b/server/pyproject.toml @@ -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 } diff --git a/server/requirements.txt b/server/requirements.txt index 1ed3c715..2dd3e2d6 100644 --- a/server/requirements.txt +++ b/server/requirements.txt @@ -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"