summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/HuggingFace.py
diff options
context:
space:
mode:
authorHeiner Lohaus <hlohaus@users.noreply.github.com>2024-04-08 07:24:00 +0200
committerHeiner Lohaus <hlohaus@users.noreply.github.com>2024-04-08 07:24:00 +0200
commit926ddfd5438ca06840bdff9b9bd21ddcd8863aa9 (patch)
treeb85cf6df7365e5800d9417ce14b010dc0142d727 /g4f/Provider/HuggingFace.py
parentAdd nodriver to OpenaiChat (diff)
downloadgpt4free-926ddfd5438ca06840bdff9b9bd21ddcd8863aa9.tar
gpt4free-926ddfd5438ca06840bdff9b9bd21ddcd8863aa9.tar.gz
gpt4free-926ddfd5438ca06840bdff9b9bd21ddcd8863aa9.tar.bz2
gpt4free-926ddfd5438ca06840bdff9b9bd21ddcd8863aa9.tar.lz
gpt4free-926ddfd5438ca06840bdff9b9bd21ddcd8863aa9.tar.xz
gpt4free-926ddfd5438ca06840bdff9b9bd21ddcd8863aa9.tar.zst
gpt4free-926ddfd5438ca06840bdff9b9bd21ddcd8863aa9.zip
Diffstat (limited to 'g4f/Provider/HuggingFace.py')
-rw-r--r--g4f/Provider/HuggingFace.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/g4f/Provider/HuggingFace.py b/g4f/Provider/HuggingFace.py
index 647780fd..6a05c26e 100644
--- a/g4f/Provider/HuggingFace.py
+++ b/g4f/Provider/HuggingFace.py
@@ -13,6 +13,10 @@ class HuggingFace(AsyncGeneratorProvider, ProviderModelMixin):
url = "https://huggingface.co/chat"
working = True
supports_message_history = True
+ models = [
+ "mistralai/Mixtral-8x7B-Instruct-v0.1",
+ "mistralai/Mistral-7B-Instruct-v0.2"
+ ]
default_model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
@classmethod
@@ -29,7 +33,7 @@ class HuggingFace(AsyncGeneratorProvider, ProviderModelMixin):
temperature: float = 0.7,
**kwargs
) -> AsyncResult:
- model = cls.get_model(model)
+ model = cls.get_model(model) if not model else model
headers = {}
if api_key is not None:
headers["Authorization"] = f"Bearer {api_key}"