Skip to content

Commit bc37689

Browse files
committed
Add ImagenGenerationConfig initializer with all params
1 parent f298f18 commit bc37689

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

FirebaseVertexAI/Sources/Types/Public/Imagen/ImagenGenerationConfig.swift

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,5 +20,15 @@ public struct ImagenGenerationConfig {
2020
public var imageFormat: ImagenImageFormat?
2121
public var addWatermark: Bool?
2222

23-
public init() {}
23+
public init(numberOfImages: Int? = nil,
24+
negativePrompt: String? = nil,
25+
aspectRatio: ImagenAspectRatio? = nil,
26+
imageFormat: ImagenImageFormat? = nil,
27+
addWatermark: Bool? = nil) {
28+
self.numberOfImages = numberOfImages
29+
self.negativePrompt = negativePrompt
30+
self.aspectRatio = aspectRatio
31+
self.imageFormat = imageFormat
32+
self.addWatermark = addWatermark
33+
}
2434
}

0 commit comments

Comments
 (0)