summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/HuggingChat.py
diff options
context:
space:
mode:
authorEien Mojiki 🍕 <95165750+eienmojiki206@users.noreply.github.com>2024-06-07 07:56:50 +0200
committerGitHub <noreply@github.com>2024-06-07 07:56:50 +0200
commit0850c81118e72246c1aa030dc3f7495b3a0fbb54 (patch)
tree93f1129d6d37b6f72d144c200872f391f0136aae /g4f/Provider/HuggingChat.py
parentMerge pull request #2031 from Zedai00/main (diff)
downloadgpt4free-0850c81118e72246c1aa030dc3f7495b3a0fbb54.tar
gpt4free-0850c81118e72246c1aa030dc3f7495b3a0fbb54.tar.gz
gpt4free-0850c81118e72246c1aa030dc3f7495b3a0fbb54.tar.bz2
gpt4free-0850c81118e72246c1aa030dc3f7495b3a0fbb54.tar.lz
gpt4free-0850c81118e72246c1aa030dc3f7495b3a0fbb54.tar.xz
gpt4free-0850c81118e72246c1aa030dc3f7495b3a0fbb54.tar.zst
gpt4free-0850c81118e72246c1aa030dc3f7495b3a0fbb54.zip
Diffstat (limited to 'g4f/Provider/HuggingChat.py')
-rw-r--r--g4f/Provider/HuggingChat.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/g4f/Provider/HuggingChat.py b/g4f/Provider/HuggingChat.py
index ac324390..cf2a89da 100644
--- a/g4f/Provider/HuggingChat.py
+++ b/g4f/Provider/HuggingChat.py
@@ -23,7 +23,8 @@ class HuggingChat(AsyncGeneratorProvider, ProviderModelMixin):
'NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO',
'mistralai/Mistral-7B-Instruct-v0.2',
'meta-llama/Meta-Llama-3-70B-Instruct',
- 'microsoft/Phi-3-mini-4k-instruct'
+ 'microsoft/Phi-3-mini-4k-instruct',
+ '01-ai/Yi-1.5-34B-Chat'
]
model_aliases = {
"mistralai/Mistral-7B-Instruct-v0.1": "mistralai/Mistral-7B-Instruct-v0.2"
@@ -102,7 +103,7 @@ class HuggingChat(AsyncGeneratorProvider, ProviderModelMixin):
elif line["type"] == "stream":
token = line["token"]
if first_token:
- token = token.lstrip()
+ token = token.lstrip().replace('\u0000', '')
first_token = False
yield token
elif line["type"] == "finalAnswer":