github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-24 23:54:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

One Line Fix: missing text output with huggingface TGI LLM (#7972)

Browse Source 
Small bug fix. The async _call method was missing a line to return the
generated text.

@baskaryan
This commit is contained in:
Wian Stipp 2023-07-20 06:44:29 -07:00 committed by GitHub
parent 493cbc9410
commit 0ffb7fc10c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
langchain/llms/huggingface_text_gen_inference.py
View File

@ -255,4 +255,5 @@ class HuggingFaceTextGenInference(LLM):
if not token.special: if not token.special:
if text_callback: if text_callback:
await text_callback(token.text) await text_callback(token.text)
text += token.text
return text return text