Add v1-4 and ofa-sys v0 models.

pull/24/head
Pedro Cuenca 2 years ago
parent 3eab2e33e9
commit d3c145f219

@ -59,7 +59,7 @@ extension ModelInfo {
extension ModelInfo { extension ModelInfo {
// TODO: repo does not exist yet // TODO: repo does not exist yet
static let v14Base = ModelInfo( static let v14Base = ModelInfo(
modelId: "pcuenq/coreml-stable-diffusion-v1-4", modelId: "pcuenq/coreml-stable-diffusion-1-4",
modelVersion: "CompVis/stable-diffusion-v1-4" modelVersion: "CompVis/stable-diffusion-v1-4"
) )
@ -79,11 +79,17 @@ extension ModelInfo {
supportsEncoder: true supportsEncoder: true
) )
static let ofaSmall = ModelInfo(
modelId: "pcuenq/coreml-small-stable-diffusion-v0",
modelVersion: "OFA-Sys/small-stable-diffusion-v0"
)
static let MODELS = [ static let MODELS = [
// ModelInfo.v14Base, ModelInfo.v14Base,
ModelInfo.v15Base, ModelInfo.v15Base,
ModelInfo.v2Base, ModelInfo.v2Base,
ModelInfo.v21Base ModelInfo.v21Base,
ModelInfo.ofaSmall
] ]
static func from(modelVersion: String) -> ModelInfo? { static func from(modelVersion: String) -> ModelInfo? {

Loading…
Cancel
Save