返回提交历史
Modified
g4f/Provider/Prodia.py
+12
-6
XFEstudio/gpt4free
refactor(g4f/Provider/Prodia.py): Enhance Prodia image generation parameters
fece1083
代码差异
1 个文件
+12
-6
@@ -98,6 +98,12 @@ class Prodia(AsyncGeneratorProvider, ProviderModelMixin):
98
98
model: str,
99
99
messages: Messages,
100
100
proxy: str = None,
101
negative_prompt: str = "",
102
steps: str = 20, # 1-25
103
cfg: str = 7, # 0-20
104
seed: str = "-1",
105
sampler: str = "DPM++ 2M Karras", # "Euler", "Euler a", "Heun", "DPM++ 2M Karras", "DPM++ SDE Karras", "DDIM"
106
aspect_ratio: str = "square", # "square", "portrait", "landscape"
101
107
**kwargs
102
108
) -> AsyncResult:
103
109
model = cls.get_model(model)
@@ -117,12 +123,12 @@ class Prodia(AsyncGeneratorProvider, ProviderModelMixin):
117
123
"new": "true",
118
124
"prompt": prompt,
119
125
"model": model,
120
"negative_prompt": kwargs.get("negative_prompt", ""),
121
"steps": kwargs.get("steps", 20),
122
"cfg": kwargs.get("cfg", 7),
123
"seed": kwargs.get("seed", int(time.time())),
124
"sampler": kwargs.get("sampler", "DPM++ 2M Karras"),
125
"aspect_ratio": kwargs.get("aspect_ratio", "square")
126
"negative_prompt": negative_prompt,
127
"steps": steps,
128
"cfg": cfg,
129
"seed": seed,
130
"sampler": sampler,
131
"aspect_ratio": aspect_ratio
126
132
}
127
133
128
134
async with session.get(cls.api_endpoint, params=params, proxy=proxy) as response: