diff options
| author | kqlio67 <166700875+kqlio67@users.noreply.github.com> | 2024-07-08 22:49:38 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-07-08 22:49:38 +0200 |
| commit | a8ab2ad771152e9f53fcb730caac84e5430523ff (patch) | |
| tree | 41c1c6ea45ad1a66072250ae17238dbf3cc26bdd /g4f/Provider/HuggingFace.py | |
| parent | Delete g4f/Provider/Aichatos.py (diff) | |
| download | gpt4free-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 'g4f/Provider/HuggingFace.py')
| -rw-r--r-- | g4f/Provider/HuggingFace.py | 12 |
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]" |