diff --git a/notebooks/synthid_text_huggingface_integration.ipynb b/notebooks/synthid_text_huggingface_integration.ipynb
index b2f7ed5..7925090 100644
--- a/notebooks/synthid_text_huggingface_integration.ipynb
+++ b/notebooks/synthid_text_huggingface_integration.ipynb
@@ -206,6 +206,7 @@
     "        else transformers.GPT2LMHeadModel\n",
     "    )\n",
     "    model = model_cls.from_pretrained(model_name.value, device_map='auto')\n",
+    "    model.generation_config.pad_token_id = model.generation_config.eos_token_id\n",
     "  else:\n",
     "    model_cls = (\n",
     "        synthid_mixin.SynthIDGemmaForCausalLM\n",