Skip to content

Commit f1852e6

Browse files
matthew29tangcopybara-github
authored andcommitted
feat: Add image_size field for Gemini Developer API Imagen 4 generation
PiperOrigin-RevId: 789937547
1 parent 30c0676 commit f1852e6

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

google/genai/models.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -997,7 +997,11 @@ def _GenerateImagesConfig_to_mldev(
997997
raise ValueError('add_watermark parameter is not supported in Gemini API.')
998998

999999
if getv(from_object, ['image_size']) is not None:
1000-
raise ValueError('image_size parameter is not supported in Gemini API.')
1000+
setv(
1001+
parent_object,
1002+
['parameters', 'sampleImageSize'],
1003+
getv(from_object, ['image_size']),
1004+
)
10011005

10021006
if getv(from_object, ['enhance_prompt']) is not None:
10031007
raise ValueError('enhance_prompt parameter is not supported in Gemini API.')

google/genai/tests/models/test_generate_images.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,6 @@
4444
),
4545
pytest_helper.TestTableItem(
4646
name='test_generate_size_2k_imagen4-001',
47-
exception_if_mldev='not supported in Gemini API',
4847
parameters=types._GenerateImagesParameters(
4948
model='imagen-4.0-generate-001',
5049
prompt='Red skateboard',

0 commit comments

Comments
 (0)