summaryrefslogtreecommitdiffstats
path: root/g4f/models.py
diff options
context:
space:
mode:
authorabc <98614666+xtekky@users.noreply.github.com>2024-04-13 04:09:11 +0200
committerabc <98614666+xtekky@users.noreply.github.com>2024-04-13 04:09:11 +0200
commit2fad27b2c5ce6b05591d2921c7bafefa2de7c9b5 (patch)
treeea9b462c5d963f649e89f92068254c6d6ac00501 /g4f/models.py
parentadd more models (diff)
downloadgpt4free-2fad27b2c5ce6b05591d2921c7bafefa2de7c9b5.tar
gpt4free-2fad27b2c5ce6b05591d2921c7bafefa2de7c9b5.tar.gz
gpt4free-2fad27b2c5ce6b05591d2921c7bafefa2de7c9b5.tar.bz2
gpt4free-2fad27b2c5ce6b05591d2921c7bafefa2de7c9b5.tar.lz
gpt4free-2fad27b2c5ce6b05591d2921c7bafefa2de7c9b5.tar.xz
gpt4free-2fad27b2c5ce6b05591d2921c7bafefa2de7c9b5.tar.zst
gpt4free-2fad27b2c5ce6b05591d2921c7bafefa2de7c9b5.zip
Diffstat (limited to '')
-rw-r--r--g4f/models.py11
1 files changed, 9 insertions, 2 deletions
diff --git a/g4f/models.py b/g4f/models.py
index ee8d83f8..ff8b3a9c 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -165,7 +165,7 @@ mistral_7b_v02 = Model(
mixtral_8x22b = Model(
name = "HuggingFaceH4/zephyr-orpo-141b-A35b-v0.1",
base_provider = "huggingface",
- best_provider = DeepInfra
+ best_provider = RetryProvider([HuggingChat, DeepInfra])
)
# Misc models
@@ -269,7 +269,13 @@ pi = Model(
dbrx_instruct = Model(
name = 'databricks/dbrx-instruct',
base_provider = 'mistral',
- best_provider = DeepInfra
+ best_provider = RetryProvider([DeepInfra, PerplexityLabs])
+)
+
+command_r_plus = Model(
+ name = 'CohereForAI/c4ai-command-r-plus',
+ base_provider = 'mistral',
+ best_provider = HuggingChat
)
class ModelUtils:
@@ -324,6 +330,7 @@ class ModelUtils:
'claude-3-sonnet': claude_3_sonnet,
# other
+ 'command-r+': command_r_plus,
'dbrx-instruct': dbrx_instruct,
'lzlv-70b': lzlv_70b,
'airoboros-70b': airoboros_70b,