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

Start work on more clean client.

Browse Source
This commit is contained in:
Alex Zenla
2023-04-22 16:32:54 -07:00
parent 1c0fbe02db
commit 4430bdcdd7
9 changed files with 69 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Sources/StableDiffusionServer/ModelService.swift
View File

@ -23,9 +23,7 @@ class ModelServiceProvider: SdModelServiceAsyncProvider {
}
func loadModel(request: SdLoadModelRequest, context _: GRPCAsyncServerCallContext) async throws -> SdLoadModelResponse {
guard let state = await modelManager.getModelState(name: request.modelName) else {
throw SdCoreError.modelNotFound
}
let state = try await modelManager.createModelState(name: request.modelName)
try await state.load()
return SdLoadModelResponse()
}

2
Sources/StableDiffusionServer/main.swift
View File

@ -15,13 +15,11 @@ struct ServerCommand: ParsableCommand {
let semaphore = DispatchSemaphore(value: 0)
Task {
print("Loading initial models...")
do {
try await modelManager.reloadModels()
} catch {
ServerCommand.exit(withError: error)
}
print("Loaded initial models.")
semaphore.signal()
}
semaphore.wait()