summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/HuggingFace.py
diff options
context:
space:
mode:
authorkqlio67 <166700875+kqlio67@users.noreply.github.com>2024-07-08 22:49:38 +0200
committerGitHub <noreply@github.com>2024-07-08 22:49:38 +0200
commita8ab2ad771152e9f53fcb730caac84e5430523ff (patch)
tree41c1c6ea45ad1a66072250ae17238dbf3cc26bdd /g4f/Provider/HuggingFace.py
parentDelete g4f/Provider/Aichatos.py (diff)
downloadgpt4free-a8ab2ad771152e9f53fcb730caac84e5430523ff.tar
gpt4free-a8ab2ad771152e9f53fcb730caac84e5430523ff.tar.gz
gpt4free-a8ab2ad771152e9f53fcb730caac84e5430523ff.tar.bz2
gpt4free-a8ab2ad771152e9f53fcb730caac84e5430523ff.tar.lz
gpt4free-a8ab2ad771152e9f53fcb730caac84e5430523ff.tar.xz
gpt4free-a8ab2ad771152e9f53fcb730caac84e5430523ff.tar.zst
gpt4free-a8ab2ad771152e9f53fcb730caac84e5430523ff.zip
Diffstat (limited to '')
-rw-r--r--g4f/Provider/HuggingFace.py12
1 files changed, 9 insertions, 3 deletions
diff --git a/g4f/Provider/HuggingFace.py b/g4f/Provider/HuggingFace.py
index 6a05c26e..a5e27ccf 100644
--- a/g4f/Provider/HuggingFace.py
+++ b/g4f/Provider/HuggingFace.py
@@ -12,10 +12,16 @@ from ..requests.raise_for_status import raise_for_status
class HuggingFace(AsyncGeneratorProvider, ProviderModelMixin):
url = "https://huggingface.co/chat"
working = True
+ needs_auth = True
supports_message_history = True
models = [
- "mistralai/Mixtral-8x7B-Instruct-v0.1",
- "mistralai/Mistral-7B-Instruct-v0.2"
+ 'CohereForAI/c4ai-command-r-plus',
+ 'meta-llama/Meta-Llama-3-70B-Instruct',
+ '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.2',
+ 'microsoft/Phi-3-mini-4k-instruct',
]
default_model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
@@ -74,4 +80,4 @@ def format_prompt(messages: Messages) -> str:
for idx, message in enumerate(messages)
if message["role"] == "assistant"
])
- return f"{history}<s>[INST] {question} [/INST]" \ No newline at end of file
+ return f"{history}<s>[INST] {question} [/INST]"