Skip to content

Commit

Permalink
improve tutorial
Browse files Browse the repository at this point in the history
  • Loading branch information
anakin87 committed Dec 6, 2023
1 parent e1548d4 commit a57f03b
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tutorials/24_Building_Chat_App.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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."
]
},
{
Expand All @@ -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\"])"
]
},
{
Expand Down

0 comments on commit a57f03b

Please sign in to comment.