From 275b926cf745b5668d3ea30236635e20e7866442 Mon Sep 17 00:00:00 2001
From: Bearnardd <43574448+Bearnardd@users.noreply.github.com>
Date: Sat, 15 Jul 2023 02:03:23 +0200
Subject: [PATCH] add missing import (#7730)

Just a nit documentation fix

 @baskaryan
---
 .../llms/integrations/huggingface_textgen_inference.ipynb       | 2 ++
 1 file changed, 2 insertions(+)

diff --git 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
index 8fa72d948b..72c79cee7b 100644
--- 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",