Skip to content

Commit 4876a0e

Browse files
committed
Update createImagenModel signature
1 parent 3e7a3b8 commit 4876a0e

File tree

2 files changed

+2
-5
lines changed

2 files changed

+2
-5
lines changed

FirebaseVertexAI/Sources/VertexAI+Preview.swift

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,12 +17,10 @@ import Foundation
1717

1818
@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
1919
public extension VertexAI {
20-
func imagenModel(modelName: String, modelConfig: ImagenModelConfig? = nil,
21-
safetySettings: ImagenSafetySettings? = nil,
20+
func imagenModel(modelName: String, safetySettings: ImagenSafetySettings? = nil,
2221
requestOptions: RequestOptions = RequestOptions()) -> ImagenModel {
2322
return createImagenModel(
2423
modelName: modelName,
25-
modelConfig: modelConfig,
2624
safetySettings: safetySettings,
2725
requestOptions: requestOptions
2826
)

FirebaseVertexAI/Sources/VertexAI.swift

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,8 +106,7 @@ public class VertexAI {
106106

107107
// MARK: - Package APIs
108108

109-
package func createImagenModel(modelName: String, modelConfig: ImagenModelConfig?,
110-
safetySettings: ImagenSafetySettings?,
109+
package func createImagenModel(modelName: String, safetySettings: ImagenSafetySettings?,
111110
requestOptions: RequestOptions) -> ImagenModel {
112111
return ImagenModel(
113112
name: modelResourceName(modelName: modelName),

0 commit comments

Comments
 (0)