Support for starting images and many more parameters.

This commit is contained in:
alex committed 2023-04-23 02:40:41 -07:00
1 parent 7c0b2779f4
commit d31e80bf4c
7 files changed
+176 -3

No files matched your search

@@ -11,9 +11,14 @@ class ImageGenerationServiceProvider: SdImageGenerationServiceAsyncProvider {
}
func generateImages(request: SdGenerateImagesRequest, context _: GRPCAsyncServerCallContext) async throws -> SdGenerateImagesResponse {
guard let state = await modelManager.getModelState(name: request.modelName) else {
throw SdCoreError.modelNotFound
do {
guard let state = await modelManager.getModelState(name: request.modelName) else {
throw SdCoreError.modelNotFound
}
return try await state.generate(request)
} catch {
print(error)
throw error
}
return try await state.generate(request)
}
}