summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/Prodia.py
diff options
context:
space:
mode:
authorkqlio67 <kqlio67@users.noreply.github.com>2024-11-08 20:00:43 +0100
committerkqlio67 <kqlio67@users.noreply.github.com>2024-11-08 20:00:43 +0100
commitfece1083182de072281bb09e1e24db3befcf7516 (patch)
treedc664879b80753f1599aa841b6c3a935db8e9c5d /g4f/Provider/Prodia.py
parentUpdate (g4f/models.py g4f/Provider/airforce/ g4f/Provider/Airforce.py) (diff)
downloadgpt4free-fece1083182de072281bb09e1e24db3befcf7516.tar
gpt4free-fece1083182de072281bb09e1e24db3befcf7516.tar.gz
gpt4free-fece1083182de072281bb09e1e24db3befcf7516.tar.bz2
gpt4free-fece1083182de072281bb09e1e24db3befcf7516.tar.lz
gpt4free-fece1083182de072281bb09e1e24db3befcf7516.tar.xz
gpt4free-fece1083182de072281bb09e1e24db3befcf7516.tar.zst
gpt4free-fece1083182de072281bb09e1e24db3befcf7516.zip
Diffstat (limited to '')
-rw-r--r--g4f/Provider/Prodia.py18
1 files changed, 12 insertions, 6 deletions
diff --git a/g4f/Provider/Prodia.py b/g4f/Provider/Prodia.py
index 543a8b19..fcebf7e3 100644
--- a/g4f/Provider/Prodia.py
+++ b/g4f/Provider/Prodia.py
@@ -98,6 +98,12 @@ class Prodia(AsyncGeneratorProvider, ProviderModelMixin):
model: str,
messages: Messages,
proxy: str = None,
+ negative_prompt: str = "",
+ steps: str = 20, # 1-25
+ cfg: str = 7, # 0-20
+ seed: str = "-1",
+ sampler: str = "DPM++ 2M Karras", # "Euler", "Euler a", "Heun", "DPM++ 2M Karras", "DPM++ SDE Karras", "DDIM"
+ aspect_ratio: str = "square", # "square", "portrait", "landscape"
**kwargs
) -> AsyncResult:
model = cls.get_model(model)
@@ -117,12 +123,12 @@ class Prodia(AsyncGeneratorProvider, ProviderModelMixin):
"new": "true",
"prompt": prompt,
"model": model,
- "negative_prompt": kwargs.get("negative_prompt", ""),
- "steps": kwargs.get("steps", 20),
- "cfg": kwargs.get("cfg", 7),
- "seed": kwargs.get("seed", int(time.time())),
- "sampler": kwargs.get("sampler", "DPM++ 2M Karras"),
- "aspect_ratio": kwargs.get("aspect_ratio", "square")
+ "negative_prompt": negative_prompt,
+ "steps": steps,
+ "cfg": cfg,
+ "seed": seed,
+ "sampler": sampler,
+ "aspect_ratio": aspect_ratio
}
async with session.get(cls.api_endpoint, params=params, proxy=proxy) as response: