diff --git a/docs/source-pytorch/conf.py b/docs/source-pytorch/conf.py
index 1488a3b625d04..1a9b15f5cc4ea 100644
--- a/docs/source-pytorch/conf.py
+++ b/docs/source-pytorch/conf.py
@@ -90,7 +90,8 @@ def _load_py_module(name: str, location: str) -> ModuleType:
 assist_local.AssistantCLI.pull_docs_files(
     gh_user_repo="Lightning-AI/lightning-Habana",
     target_dir="docs/source-pytorch/integrations/hpu",
-    checkout="refs/tags/1.6.0",
+    # checkout="refs/tags/1.6.0",
+    checkout="5549fa927d5501d31aac0c9b2ed479be62a02cbc",
 )
 # the HPU also need some images
 URL_RAW_DOCS_HABANA = "https://raw.githubusercontent.com/Lightning-AI/lightning-Habana/1.5.0/docs/source"