Archives/gpt4free
2
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-11-10 19:11:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(models): update HuggingChat, HuggingFace provider models and aliases

Browse Source
This commit is contained in:
kqlio67 2024-09-12 18:06:52 +03:00
parent abefd7f36f
commit 11f702e0ac
3 changed files with 2 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
g4f/Provider/HuggingChat.py
View File

@ -13,11 +13,10 @@ class HuggingChat(AbstractProvider, ProviderModelMixin):
supports_stream = True
default_model = "meta-llama/Meta-Llama-3.1-70B-Instruct"
models = [
'meta-llama/Meta-Llama-3.1-70B-Instruct',
default_model,
'CohereForAI/c4ai-command-r-plus-08-2024',
'mistralai/Mixtral-8x7B-Instruct-v0.1',
'NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO',
'01-ai/Yi-1.5-34B-Chat',
'mistralai/Mistral-7B-Instruct-v0.3',
'microsoft/Phi-3-mini-4k-instruct',
]
@ -27,7 +26,6 @@ class HuggingChat(AbstractProvider, ProviderModelMixin):
"command-r-plus": "CohereForAI/c4ai-command-r-plus-08-2024",
"mixtral-8x7b": "mistralai/Mixtral-8x7B-Instruct-v0.1",
"mixtral-8x7b": "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
"yi-1.5-34b": "01-ai/Yi-1.5-34B-Chat",
"mistral-7b": "mistralai/Mistral-7B-Instruct-v0.3",
"phi-3-mini-4k": "microsoft/Phi-3-mini-4k-instruct",
}

4
g4f/Provider/HuggingFace.py
View File

@ -16,11 +16,10 @@ class HuggingFace(AsyncGeneratorProvider, ProviderModelMixin):
supports_message_history = True
default_model = "meta-llama/Meta-Llama-3.1-70B-Instruct"
models = [
'meta-llama/Meta-Llama-3.1-70B-Instruct',
default_model,
'CohereForAI/c4ai-command-r-plus-08-2024',
'mistralai/Mixtral-8x7B-Instruct-v0.1',
'NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO',
'01-ai/Yi-1.5-34B-Chat',
'mistralai/Mistral-7B-Instruct-v0.3',
'microsoft/Phi-3-mini-4k-instruct',
]
@ -30,7 +29,6 @@ class HuggingFace(AsyncGeneratorProvider, ProviderModelMixin):
"command-r-plus": "CohereForAI/c4ai-command-r-plus-08-2024",
"mixtral-8x7b": "mistralai/Mixtral-8x7B-Instruct-v0.1",
"mixtral-8x7b": "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
"yi-1.5-34b": "01-ai/Yi-1.5-34B-Chat",
"mistral-7b": "mistralai/Mistral-7B-Instruct-v0.3",
"phi-3-mini-4k": "microsoft/Phi-3-mini-4k-instruct",
}

11
g4f/models.py
View File

@ -202,13 +202,6 @@ mistral_7b = Model(
best_provider = IterListProvider([HuggingChat, HuggingFace, DeepInfra])
)
### 01-ai ###
yi_1_5_34b = Model(
name = "yi-1.5-34b",
base_provider = "01-ai",
best_provider = IterListProvider([HuggingChat, HuggingFace])
)
### Microsoft ###
phi_3_mini_4k = Model(
@ -550,10 +543,6 @@ class ModelUtils:
'mistral-7b': mistral_7b,
### 01-ai ###
'yi-1.5-34b': yi_1_5_34b,
### Microsoft ###
'phi-3-mini-4k': phi_3_mini_4k,