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

18
Sources/StableDiffusionCore/ModelManager.swift
View File

@ -15,6 +15,7 @@ public actor ModelManager {
public func reloadModels() throws {
modelInfos.removeAll()
modelUrls.removeAll()
modelStates.removeAll()
let contents = try FileManager.default.contentsOfDirectory(at: modelBaseURL.resolvingSymlinksInPath(), includingPropertiesForKeys: [.isDirectoryKey])
for subdirectoryURL in contents {
@ -29,6 +30,22 @@ public actor ModelManager {
Array(modelInfos.values)
}
public func createModelState(name: String) throws -> ModelState {
let state = modelStates[name]
let url = modelUrls[name]
guard let url else {
throw SdCoreError.modelNotFound
}
if state == nil {
let state = ModelState(url: url)
modelStates[name] = state
return state
} else {
return state!
}
}
public func getModelState(name: String) -> ModelState? {
modelStates[name]
}
@ -40,7 +57,6 @@ public actor ModelManager {
info.attention = attention ?? "unknown"
modelInfos[info.name] = info
modelUrls[info.name] = url
modelStates[info.name] = try ModelState(url: url)
}
private func getModelAttention(_ url: URL) -> String? {