Skip to content

Commit 495a64b

Browse files
authored
feat: add parameter for output image resolution (#13519)
* feat: add parameter for output image resolution * fix: spacing * fix: update genai version * fix: add config
1 parent a09a2a9 commit 495a64b

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

genai/image_generation/imggen_with_txt.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
def generate_images(output_file: str) -> Image:
1919
# [START googlegenaisdk_imggen_with_txt]
2020
from google import genai
21+
from google.genai.types import GenerateImagesConfig
2122

2223
client = genai.Client()
2324

@@ -27,6 +28,9 @@ def generate_images(output_file: str) -> Image:
2728
image = client.models.generate_images(
2829
model="imagen-4.0-generate-preview-06-06",
2930
prompt="A dog reading a newspaper",
31+
config=GenerateImagesConfig(
32+
image_size="2K",
33+
),
3034
)
3135

3236
image.generated_images[0].image.save(output_file)

0 commit comments

Comments
 (0)