diff --git a/docs/modules/llms/integrations/huggingface_hub.ipynb b/docs/modules/llms/integrations/huggingface_hub.ipynb index 5e4769f75ce..803f67ecba8 100644 --- a/docs/modules/llms/integrations/huggingface_hub.ipynb +++ b/docs/modules/llms/integrations/huggingface_hub.ipynb @@ -20,7 +20,7 @@ "name": "stdout", "output_type": "stream", "text": [ - "The Seattle Seahawks won the Super Bowl in 2010. Justin Beiber was born in 2010. The\n" + "The Seattle Seahawks won the Super Bowl in 2010. Justin Beiber was born in 2010. The final answer: Seattle Seahawks.\n" ] } ], @@ -31,7 +31,7 @@ "\n", "Answer: Let's think step by step.\"\"\"\n", "prompt = PromptTemplate(template=template, input_variables=[\"question\"])\n", - "llm_chain = LLMChain(prompt=prompt, llm=HuggingFaceHub(repo_id=\"google/flan-t5-xl\", model_kwargs={\"temperature\":1e-10}))\n", + "llm_chain = LLMChain(prompt=prompt, llm=HuggingFaceHub(repo_id=\"google/flan-t5-xl\", model_kwargs={\"temperature\":0, \"max_length\":64}))\n", "\n", "question = \"What NFL team won the Super Bowl in the year Justin Beiber was born?\"\n", "\n",