diff --git a/docs/source-pytorch/conf.py b/docs/source-pytorch/conf.py
index 107f58c579..be5bbd07e6 100644
--- a/docs/source-pytorch/conf.py
+++ b/docs/source-pytorch/conf.py
@@ -90,7 +90,7 @@ _transform_changelog(
 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="1.7.0.rc0",
 )
 # the HPU also need some images
 URL_RAW_DOCS_HABANA = "https://raw.githubusercontent.com/Lightning-AI/lightning-Habana/1.5.0/docs/source"