add missing import (#7730)

Just a nit documentation fix @baskaryan
1 year ago · 275b926cf7
parent 9800c6051c
commit 275b926cf7
1 changed files with 2 additions and 0 deletions
--- a/docs/extras/modules/model_io/models/llms/integrations/huggingface_textgen_inference.ipynb
+++ b/docs/extras/modules/model_io/models/llms/integrations/huggingface_textgen_inference.ipynb
@ -35,6 +35,8 @@
   "metadata": {},
   "outputs": [],
   "source": [
+    "from langchain.llms import HuggingFaceTextGenInference\n",
+    "\n",
    "llm = HuggingFaceTextGenInference(\n",
    "    inference_server_url=\"http://localhost:8010/\",\n",
    "    max_new_tokens=512,\n",