strips/stable-diffusion-rpc
2
0
Fork 0
mirror of https://github.com/GayPizzaSpecifications/stable-diffusion-rpc.git synced 2025-08-04 14:01:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support for starting images and many more parameters.

Browse Source
This commit is contained in:
Alex Zenla
2023-04-23 02:40:41 -07:00
parent 7c0b2779f4
commit d31e80bf4c
7 changed files with 176 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
Sources/StableDiffusionServer/ImageGenerationService.swift
View File

@ -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)
}
}