fix: hugginface

2023-09-23 08:01:21 +02:00 · 2023-09-23 08:01:21 +02:00 · d869e11eb6
parent fe1db5ab13
commit d869e11eb6
1 changed files with 2 additions and 1 deletions
--- a/src/providers/hfbasechat.py
+++ b/src/providers/hfbasechat.py
@ -32,8 +32,9 @@ class BaseHFChatProvider(BaseProvider):
                },
            })
        else:
+            prompt = self.make_prompt(prompt, chat)
            output = query({
-                "inputs": self.make_prompt(prompt, chat),
+                "inputs": prompt,
            })

        if 'generated_text' in output: