diff --git a/server/pyproject.toml b/server/pyproject.toml
index 46a513117..62c52c7b1 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.14.1"
+#optimum-habana = "1.15.0"
 transformers = "4.45.2"
 numpy = "1.26.4"
 accelerate = "0.33.0"
diff --git a/server/requirements.txt b/server/requirements.txt
index a940574f9..11e51693b 100644
--- a/server/requirements.txt
+++ b/server/requirements.txt
@@ -46,7 +46,8 @@ 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==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==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"