summaryrefslogtreecommitdiffstats
path: root/g4f/models.py
diff options
context:
space:
mode:
authornullstreak <139914347+nullstreak@users.noreply.github.com>2024-04-12 05:18:08 +0200
committernullstreak <139914347+nullstreak@users.noreply.github.com>2024-04-12 05:18:08 +0200
commit2e171fc290cee3c32dc759ee9037bbaf2c045e74 (patch)
treeccb2b5cb350d359afe052e090eed597ca34c63cf /g4f/models.py
parentUpdate chat.v1.js (diff)
downloadgpt4free-2e171fc290cee3c32dc759ee9037bbaf2c045e74.tar
gpt4free-2e171fc290cee3c32dc759ee9037bbaf2c045e74.tar.gz
gpt4free-2e171fc290cee3c32dc759ee9037bbaf2c045e74.tar.bz2
gpt4free-2e171fc290cee3c32dc759ee9037bbaf2c045e74.tar.lz
gpt4free-2e171fc290cee3c32dc759ee9037bbaf2c045e74.tar.xz
gpt4free-2e171fc290cee3c32dc759ee9037bbaf2c045e74.tar.zst
gpt4free-2e171fc290cee3c32dc759ee9037bbaf2c045e74.zip
Diffstat (limited to '')
-rw-r--r--g4f/models.py25
1 files changed, 16 insertions, 9 deletions
diff --git a/g4f/models.py b/g4f/models.py
index 7be74dbf..875f900a 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -127,7 +127,7 @@ llama2_70b = Model(
codellama_34b_instruct = Model(
name = "codellama/CodeLlama-34b-Instruct-hf",
base_provider = "meta",
- best_provider = RetryProvider([HuggingChat, DeepInfra])
+ best_provider = HuggingChat
)
codellama_70b_instruct = Model(
@@ -146,9 +146,21 @@ mixtral_8x7b = Model(
mistral_7b = Model(
name = "mistralai/Mistral-7B-Instruct-v0.1",
base_provider = "huggingface",
- best_provider = RetryProvider([DeepInfra, HuggingChat, HuggingFace, PerplexityLabs])
+ best_provider = RetryProvider([HuggingChat, HuggingFace, PerplexityLabs])
+)
+
+mistral_7b_v02 = Model(
+ name = "mistralai/Mistral-7B-Instruct-v0.2",
+ base_provider = "huggingface",
+ best_provider = DeepInfra
)
+# mixtral_8x22b = Model(
+# name = "mistralai/Mixtral-8x22B-v0.1",
+# base_provider = "huggingface",
+# best_provider = DeepInfra
+# )
+
# Misc models
dolphin_mixtral_8x7b = Model(
name = "cognitivecomputations/dolphin-2.6-mixtral-8x7b",
@@ -168,12 +180,6 @@ airoboros_70b = Model(
best_provider = DeepInfra
)
-airoboros_l2_70b = Model(
- name = "jondurbin/airoboros-l2-70b-gpt4-1.4.1",
- base_provider = "huggingface",
- best_provider = DeepInfra
-)
-
openchat_35 = Model(
name = "openchat/openchat_3.5",
base_provider = "huggingface",
@@ -290,10 +296,11 @@ class ModelUtils:
'mixtral-8x7b': mixtral_8x7b,
'mistral-7b': mistral_7b,
+ 'mistral-7b-v02': mistral_7b_v02,
+ # 'mixtral-8x22b': mixtral_8x22b,
'dolphin-mixtral-8x7b': dolphin_mixtral_8x7b,
'lzlv-70b': lzlv_70b,
'airoboros-70b': airoboros_70b,
- 'airoboros-l2-70b': airoboros_l2_70b,
'openchat_3.5': openchat_35,
'gemini': gemini,
'gemini-pro': gemini_pro,