返回提交历史
Modified
g4f/Provider/airforce/AirforceImage.py
+8
-14
XFEstudio/gpt4free
feat(g4f/Provider/airforce/AirforceImage.py): Dynamically fetch image models from API
562a5c95
代码差异
1 个文件
+8
-14
@@ -3,6 +3,7 @@ from __future__ import annotations
3
3
from aiohttp import ClientSession
4
4
from urllib.parse import urlencode
5
5
import random
6
import requests
6
7
7
8
from ...typing import AsyncResult, Messages
8
9
from ...image import ImageResponse
@@ -16,20 +17,13 @@ class AirforceImage(AsyncGeneratorProvider, ProviderModelMixin):
16
17
#working = True
17
18
18
19
default_model = 'flux'
19
image_models = [
20
'flux',
21
'flux-realism',
22
'flux-anime',
23
'flux-3d',
24
'flux-disney',
25
'flux-pixel',
26
'flux-4o',
27
'any-dark',
28
'stable-diffusion-xl-base',
29
'stable-diffusion-xl-lightning',
30
'Flux-1.1-Pro',
31
]
32
models = [*image_models]
20
21
response = requests.get('https://api.airforce/imagine/models')
22
data = response.json()
23
24
image_models = data
25
26
models = [*image_models, "stable-diffusion-xl-base", "stable-diffusion-xl-lightning", "Flux-1.1-Pro"]
33
27
34
28
model_aliases = {
35
29
"sdxl": "stable-diffusion-xl-base",