From 769c3bb83842af20482de261ddc24b03f861755b Mon Sep 17 00:00:00 2001 From: Michal Gregor Date: Thu, 1 Aug 2024 17:42:35 +0200 Subject: [PATCH] huggingface: Added a missing argument to a ChatHuggingFace doc notebook. (#24929) - **Description:** When adding docs for constructing ChatHuggingFace using a HuggingFacePipeline, I forgot to add `return_full_text=False` as an argument. In this setup, the chat response would incorrectly contain all the input text. I am fixing that here by adding that line to the offending notebook. --------- Co-authored-by: Chester Curme --- docs/docs/integrations/chat/huggingface.ipynb | 1 + 1 file changed, 1 insertion(+) diff --git a/docs/docs/integrations/chat/huggingface.ipynb b/docs/docs/integrations/chat/huggingface.ipynb index 5fb4d7df872..3dc23997c4c 100644 --- a/docs/docs/integrations/chat/huggingface.ipynb +++ b/docs/docs/integrations/chat/huggingface.ipynb @@ -129,6 +129,7 @@ " max_new_tokens=512,\n", " do_sample=False,\n", " repetition_penalty=1.03,\n", + " return_full_text=False,\n", " ),\n", ")" ]