mirrors/gpt4free
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-06 02:30:41 -08:00
Code Issues Projects Releases Packages Wiki Activity

Update HuggingChat provider

Browse source 
- Fix null string in response token temporarily 
- Add new model `01-ai/Yi-1.5-34B-Chat` that works well using Japanese.
This commit is contained in:
Eien Mojiki 🍕 2024-06-07 12:56:50 +07:00 • committed by GitHub
parent 87a96ce0d7
commit 0850c81118
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
g4f/Provider/HuggingChat.py
View file

@ -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":