返回提交历史
Modified
g4f/Provider/PollinationsAI.py
+7
-3
XFEstudio/gpt4free
Swap model aliases for consistency and enhance model retrieval logic
9bed8da3
代码差异
1 个文件
+7
-3
@@ -88,7 +88,7 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
88
88
vision_models = [default_vision_model]
89
89
_models_loaded = False
90
90
model_aliases = {
91
"openai-fast": "gpt-4.1-nano",
91
"gpt-4.1-nano": "openai-fast",
92
92
"llama-4-scout": "llamascout",
93
93
"deepseek-r1": "deepseek-reasoning",
94
94
"sdxl-turbo": "turbo",
@@ -99,6 +99,7 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
99
99
"flux": "flux",
100
100
"flux-kontext": "kontext",
101
101
}
102
swap_model_aliases = {v: k for k, v in model_aliases.items()}
102
103
103
104
@classmethod
104
105
def get_models(cls, **kwargs):
@@ -106,8 +107,8 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
106
107
alias = model.get("name")
107
108
if (model.get("aliases")):
108
109
alias = model.get("aliases")[0]
109
elif alias in cls.model_aliases:
110
alias = cls.model_aliases[alias]
110
elif alias in cls.swap_model_aliases:
111
alias = cls.swap_model_aliases[alias]
111
112
return alias.replace("-instruct", "").replace("qwen-", "qwen").replace("qwen", "qwen-")
112
113
if not cls._models_loaded:
113
114
try:
@@ -383,6 +384,9 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
383
384
if cls.login_url in str(item):
384
385
raise MissingAuthError(item)
385
386
raise item
387
else:
388
finished += 1
389
yield Reasoning(label=f"Image {finished}/{n} failed after {time.time() - start:.2f}s: {item}")
386
390
else:
387
391
finished += 1
388
392
yield Reasoning(label=f"Image {finished}/{n} generated in {time.time() - start:.2f}s")