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

Document API, make the implementation match the API, and update the same.

Browse Source
This commit is contained in:
Alex Zenla
2023-04-23 02:09:50 -07:00
parent 71afed326f
commit 7c0b2779f4
15 changed files with 707 additions and 310 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
Sources/StableDiffusionProtos/Utilities.swift Normal file
View File

@ -0,0 +1,26 @@
import CoreML
import Foundation
public extension SdComputeUnits {
func toMlComputeUnits() -> MLComputeUnits {
switch self {
case .all: return .all
case .cpu: return .cpuOnly
case .cpuAndGpu: return .cpuAndGPU
case .cpuAndNeuralEngine: return .cpuAndNeuralEngine
default: return .all
}
}
}
public extension MLComputeUnits {
func toSdComputeUnits() -> SdComputeUnits {
switch self {
case .all: return .all
case .cpuOnly: return .cpu
case .cpuAndGPU: return .cpuAndGpu
case .cpuAndNeuralEngine: return .cpuAndNeuralEngine
default: return .all
}
}
}