From 776d7ec18e35295de2076070149c254c5bc1c10b Mon Sep 17 00:00:00 2001
From: hlohaus <983577+hlohaus@users.noreply.github.com>
Date: Sun, 9 Feb 2025 17:25:22 +0100
Subject: Add vision models to HuggingFace provider

---
 g4f/Provider/hf/__init__.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'g4f/Provider/hf/__init__.py')

diff --git a/g4f/Provider/hf/__init__.py b/g4f/Provider/hf/__init__.py
index 260f02df..a83ae3fb 100644
--- a/g4f/Provider/hf/__init__.py
+++ b/g4f/Provider/hf/__init__.py
@@ -9,7 +9,7 @@ from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
 from .HuggingChat import HuggingChat
 from .HuggingFaceAPI import HuggingFaceAPI
 from .HuggingFaceInference import HuggingFaceInference
-from .models import model_aliases
+from .models import model_aliases, vision_models, default_vision_model
 from ... import debug
 
 class HuggingFace(AsyncGeneratorProvider, ProviderModelMixin):
@@ -26,6 +26,8 @@ class HuggingFace(AsyncGeneratorProvider, ProviderModelMixin):
         return cls.models
 
     model_aliases = model_aliases
+    vision_models = vision_models
+    default_vision_model = default_vision_model
 
     @classmethod
     async def create_async_generator(
-- 
cgit v1.2.3