From d887e1140b5130ec522eb2a7fddd9d0dc0e507bb Mon Sep 17 00:00:00 2001
From: Surbhi Jain <sjsurbhi@google.com>
Date: Thu, 13 Aug 2026 05:04:25 +0000
Subject: [PATCH] Update HF login in jupyter notebooks

---
 src/maxtext/examples/sft_llama3_demo_tpu.ipynb | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/maxtext/examples/sft_llama3_demo_tpu.ipynb b/src/maxtext/examples/sft_llama3_demo_tpu.ipynb
index d4fba92eed..9bb02d5bc9 100644
--- a/src/maxtext/examples/sft_llama3_demo_tpu.ipynb
+++ b/src/maxtext/examples/sft_llama3_demo_tpu.ipynb
@@ -155,7 +155,11 @@
     "    notebook_login()\n",
     "else:\n",
     "    from huggingface_hub import login\n",
-    "    login()"
+    "    hf_token = os.environ.get(\"HF_TOKEN\", \"\")\n",
+    "    if hf_token:\n",
+    "        login(token=hf_token)\n",
+    "    else:\n",
+    "        login()"
    ]
   },
   {