From a57f03b2261b8614efc4217011c2cbf0fdf5f4ab Mon Sep 17 00:00:00 2001 From: anakin87 Date: Wed, 6 Dec 2023 08:58:11 +0100 Subject: [PATCH] improve tutorial --- tutorials/24_Building_Chat_App.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tutorials/24_Building_Chat_App.ipynb b/tutorials/24_Building_Chat_App.ipynb index ebac94cf..e9652b8c 100644 --- a/tutorials/24_Building_Chat_App.ipynb +++ b/tutorials/24_Building_Chat_App.ipynb @@ -143,7 +143,7 @@ "source": [ "### 2) Create a PromptNode\n", "\n", - "You'll initialize a PromptNode with the `model_name`, `api_key`, and `max_length` to control the output length of the model. In this tutorial, you'll use [OpenAssistant/oasst-sft-1-pythia-12b](https://huggingface.co/OpenAssistant/oasst-sft-1-pythia-12b), an open source Transformer-based text generation model." + "You'll initialize a PromptNode with the `model_name`, `api_key`, and `max_length` to control the output length of the model. In this tutorial, you'll use [HuggingFaceH4/zephyr-7b-beta](https://huggingface.co/HuggingFaceH4/zephyr-7b-beta), an open source chat Language Model." ] }, { @@ -156,8 +156,8 @@ "source": [ "from haystack.nodes import PromptNode\n", "\n", - "model_name = \"OpenAssistant/oasst-sft-1-pythia-12b\"\n", - "prompt_node = PromptNode(model_name, api_key=model_api_key, max_length=256)" + "model_name = \"https://huggingface.co/HuggingFaceH4/zephyr-7b-beta\"\n", + "prompt_node = PromptNode(model_name, api_key=model_api_key, max_length=256, stop_words=[\"Human\"])" ] }, {