Bavarder/Bavarder
Bavarder/Bavarder
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: hugginface

Browse Source
This commit is contained in:
0xMRTT 2023-09-23 08:01:21 +02:00
parent fe1db5ab13
commit d869e11eb6
No known key found for this signature in database
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/providers/hfbasechat.py
View File

@ -32,8 +32,9 @@ class BaseHFChatProvider(BaseProvider):
}, },
}) })
else: else:
prompt = self.make_prompt(prompt, chat)
output = query({ output = query({
"inputs": self.make_prompt(prompt, chat), "inputs": prompt,
}) })
if 'generated_text' in output: if 'generated_text' in output: