mirror of
https://github.com/GayPizzaSpecifications/stable-diffusion-rpc.git
synced 2025-08-03 05:30:54 +00:00
19 lines
592 B
Swift
19 lines
592 B
Swift
import Foundation
|
|
import GRPC
|
|
import StableDiffusionCore
|
|
import StableDiffusionProtos
|
|
|
|
final class HostModelServiceProvider: SdHostModelServiceAsyncProvider {
|
|
private let modelManager: ModelManager
|
|
|
|
init(modelManager: ModelManager) {
|
|
self.modelManager = modelManager
|
|
}
|
|
|
|
func loadModel(request: SdLoadModelRequest, context _: GRPCAsyncServerCallContext) async throws -> SdLoadModelResponse {
|
|
let state = try await modelManager.createModelState(name: request.modelName)
|
|
try await state.load(request: request)
|
|
return SdLoadModelResponse()
|
|
}
|
|
}
|