PrivateCoffee/matrix-gptbot
4
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: Improve image generation quality selection

Browse source 
The code change adds a conditional statement to select the image generation quality based on the selected image model.
This commit is contained in:
Kumi 2023-11-19 15:26:09 +01:00
parent 474af54ae1
commit 14da88de8b
Signed by: kumi
GPG key ID: ECBCC9082395383F
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
src/gptbot/classes/openai.py
View file

@ -228,6 +228,7 @@ Only the event_types mentioned above are allowed, you must not respond in any ot
image_partial = partial( image_partial = partial(
self.openai_api.images.generate, self.openai_api.images.generate,
model=self.image_model, model=self.image_model,
quality="standard" if self.image_model != "dall-e-3" else "hd",
prompt=prompt, prompt=prompt,
n=1, n=1,
size="1024x1024", size="1024x1024",