From 20e84145f5e4212b553e00bfbb689ad5dd173d36 Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Sat, 14 Dec 2024 19:50:57 +0100 Subject: [PATCH] Add more chat templates to HuggingFace --- g4f/Provider/needs_auth/HuggingFace.py | 1 + 1 file changed, 1 insertion(+) diff --git a/g4f/Provider/needs_auth/HuggingFace.py b/g4f/Provider/needs_auth/HuggingFace.py index 2c4607001e6..3fa5a624074 100644 --- a/g4f/Provider/needs_auth/HuggingFace.py +++ b/g4f/Provider/needs_auth/HuggingFace.py @@ -27,6 +27,7 @@ def get_models(cls) -> list[str]: url = "https://huggingface.co/api/models?inference=warm&pipeline_tag=text-generation" cls.models = [model["id"] for model in requests.get(url).json()] cls.models.append("meta-llama/Llama-3.2-11B-Vision-Instruct") + cls.models.append("nvidia/Llama-3.1-Nemotron-70B-Instruct-HF") if not cls.image_models: url = "https://huggingface.co/api/models?pipeline_tag=text-to-image" cls.image_models = [model["id"] for model in requests.get(url).json() if model["trendingScore"] >= 20]