diff --git a/optimum/habana/transformers/training_args.py b/optimum/habana/transformers/training_args.py
index 740e682f75..194a610468 100644
--- a/optimum/habana/transformers/training_args.py
+++ b/optimum/habana/transformers/training_args.py
@@ -651,12 +651,9 @@ def _setup_devices(self) -> "torch.device":
 
             if self.use_lazy_mode:
                 logger.info("Enabled lazy mode.")
-            # TODO: remove the block below when upgrade to SynapseAI 1.13 is done
-            # as eager mode will not be available anymore
             elif not self.torch_compile:
                 if os.getenv("PT_HPU_LAZY_MODE", "1") != "0":
-                    os.environ["PT_HPU_LAZY_MODE"] = "2"
-                logger.info("Enabled eager mode because use_lazy_mode=False.")
+                    assert False, "Lazy mode or compile mode not enabled => eager mode should be enabled using PT_HPU_LAZY_MODE=0"
 
             if self.deepspeed:
                 # Need to do similar for Accelerator init
diff --git a/optimum/habana/version.py b/optimum/habana/version.py
index bed1b4a432..09f2149846 100644
--- a/optimum/habana/version.py
+++ b/optimum/habana/version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "1.9.0.dev0"
+__version__ = "1.9.0"
diff --git a/setup.py b/setup.py
index 100348d060..3ab10a2220 100644
--- a/setup.py
+++ b/setup.py
@@ -49,7 +49,7 @@
 
 QUALITY_REQUIRES = [
     "ruff",
-    "hf_doc_builder @ git+https://github.com/huggingface/doc-builder.git",
+    "hf_doc_builder",
 ]
 
 EXTRAS_REQUIRE = {