pull/21914/head
keenborder786 2 weeks ago
parent e918015cf8
commit 8e233a3be0

@ -131,7 +131,7 @@
"repo_id = \"mistralai/Mistral-7B-Instruct-v0.2\"\n",
"\n",
"llm = HuggingFaceEndpoint(\n",
" repo_id=repo_id, max_length=128, temperature=0.5, token=HUGGINGFACEHUB_API_TOKEN\n",
" repo_id=repo_id, max_length=128, temperature=0.5, huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN\n",
")\n",
"llm_chain = prompt | llm\n",
"print(llm_chain.invoke({'question':question}))"

Loading…
Cancel
Save