Add more chat templates to HuggingFace

hlohaus · Dec 14, 2024 · 20e8414 · 20e8414
1 parent 3bab11c
commit 20e8414
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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]