返回提交历史
Modified
g4f/Provider/hf/HuggingFaceAPI.py
+4
-4
Modified
g4f/Provider/hf/__init__.py
+3
-1
Modified
g4f/Provider/hf/models.py
+3
-1
Modified
g4f/Provider/hf_space/G4F.py
+2
-2
XFEstudio/gpt4free
Add vision models to HuggingFace provider
776d7ec1
代码差异
4 个文件
+12
-8
@@ -5,7 +5,7 @@ from ...typing import ImagesType
5
5
from ...requests import StreamSession, raise_for_status
6
6
from ...errors import ModelNotSupportedError
7
7
from ..template.OpenaiTemplate import OpenaiTemplate
8
from .models import model_aliases
8
from .models import model_aliases, vision_models, default_vision_model
9
9
from .HuggingChat import HuggingChat
10
10
from ... import debug
11
11
@@ -17,9 +17,9 @@ class HuggingFaceAPI(OpenaiTemplate):
17
17
working = True
18
18
needs_auth = True
19
19
20
default_model = "meta-llama/Llama-3.2-11B-Vision-Instruct"
21
default_vision_model = default_model
22
vision_models = [default_vision_model, "Qwen/Qwen2-VL-7B-Instruct"]
20
default_model = default_vision_model
21
default_vision_model = default_vision_model
22
vision_models = vision_models
23
23
model_aliases = model_aliases
24
24
25
25
@classmethod
@@ -9,7 +9,7 @@ from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
9
9
from .HuggingChat import HuggingChat
10
10
from .HuggingFaceAPI import HuggingFaceAPI
11
11
from .HuggingFaceInference import HuggingFaceInference
12
from .models import model_aliases
12
from .models import model_aliases, vision_models, default_vision_model
13
13
from ... import debug
14
14
15
15
class HuggingFace(AsyncGeneratorProvider, ProviderModelMixin):
@@ -26,6 +26,8 @@ class HuggingFace(AsyncGeneratorProvider, ProviderModelMixin):
26
26
return cls.models
27
27
28
28
model_aliases = model_aliases
29
vision_models = vision_models
30
default_vision_model = default_vision_model
29
31
30
32
@classmethod
31
33
async def create_async_generator(
@@ -43,4 +43,6 @@ extra_models = [
43
43
"meta-llama/Llama-3.2-11B-Vision-Instruct",
44
44
"nvidia/Llama-3.1-Nemotron-70B-Instruct-HF",
45
45
"NousResearch/Hermes-3-Llama-3.1-8B",
46
]
46
]
47
default_vision_model = "meta-llama/Llama-3.2-11B-Vision-Instruct"
48
vision_models = [default_vision_model, "Qwen/Qwen2-VL-7B-Instruct"]
@@ -43,7 +43,7 @@ class G4F(Janus_Pro_7B):
43
43
zerogpu_uuid: str = "[object Object]",
44
44
**kwargs
45
45
) -> AsyncResult:
46
if model == "flux-dev":
46
if model in ("flux", "flux-dev"):
47
47
async for chunk in FluxDev.create_async_generator(
48
48
model, messages,
49
49
proxy=proxy,
@@ -108,7 +108,7 @@ class G4F(Janus_Pro_7B):
108
108
await raise_for_status(response)
109
109
response_data = await response.json()
110
110
image_url = response_data["data"][0]['url']
111
return ImageResponse(images=[image_url], alt=prompt)
111
return ImageResponse(image_url, alt=prompt)
112
112
background_tasks = set()
113
113
started = time.time()
114
114
task = asyncio.create_task(generate())