File tree Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -997,7 +997,11 @@ def _GenerateImagesConfig_to_mldev(
997
997
raise ValueError ('add_watermark parameter is not supported in Gemini API.' )
998
998
999
999
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
+ )
1001
1005
1002
1006
if getv (from_object , ['enhance_prompt' ]) is not None :
1003
1007
raise ValueError ('enhance_prompt parameter is not supported in Gemini API.' )
Original file line number Diff line number Diff line change 44
44
),
45
45
pytest_helper .TestTableItem (
46
46
name = 'test_generate_size_2k_imagen4-001' ,
47
- exception_if_mldev = 'not supported in Gemini API' ,
48
47
parameters = types ._GenerateImagesParameters (
49
48
model = 'imagen-4.0-generate-001' ,
50
49
prompt = 'Red skateboard' ,
You can’t perform that action at this time.
0 commit comments