yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-10-27 07:48:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove duplicate (#5368)

Browse Source
This commit is contained in:
Fanli Lin 2023-07-17 22:24:02 +08:00 committed by GitHub
parent 8f3fe85878
commit 09a1d3c0dc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
haystack/nodes/prompt/invocation_layer/hugging_face.py
View File

@ -235,7 +235,6 @@ class HFLocalInvocationLayer(PromptModelInvocationLayer):
# Thus only generated text is returned (excluding prompt) # Thus only generated text is returned (excluding prompt)
if is_text_generation and "return_full_text" not in model_input_kwargs: if is_text_generation and "return_full_text" not in model_input_kwargs:
model_input_kwargs["return_full_text"] = False model_input_kwargs["return_full_text"] = False
model_input_kwargs["max_new_tokens"] = self.max_length
if stop_words: if stop_words:
sw = StopWordsCriteria(tokenizer=self.pipe.tokenizer, stop_words=stop_words, device=self.pipe.device) sw = StopWordsCriteria(tokenizer=self.pipe.tokenizer, stop_words=stop_words, device=self.pipe.device)
model_input_kwargs["stopping_criteria"] = StoppingCriteriaList([sw]) model_input_kwargs["stopping_criteria"] = StoppingCriteriaList([sw])