Rename context -> generation

pull/12/head
Pedro Cuenca 1 year ago
parent 82cea4f559
commit e6d725df67

@ -8,7 +8,7 @@
import SwiftUI
struct ContentView: View {
@StateObject var context = GenerationContext()
@StateObject var generation = GenerationContext()
var body: some View {
NavigationSplitView {
@ -26,7 +26,7 @@ struct ContentView: View {
}
}
.environmentObject(context)
.environmentObject(generation)
}
}

@ -18,7 +18,7 @@ enum PipelineState {
}
struct PromptView: View {
@EnvironmentObject var context: GenerationContext
@EnvironmentObject var generation: GenerationContext
static let models = ModelInfo.MODELS
static let modelNames = models.map { $0.modelVersion }
@ -57,7 +57,7 @@ struct PromptView: View {
}
}
do {
context.pipeline = try await loader.prepare()
generation.pipeline = try await loader.prepare()
pipelineState = .ready
} catch {
print("Could not load model, error: \(error)")

@ -12,7 +12,7 @@ import Combine
let model = ModelInfo.v2Base
struct LoadingView: View {
@StateObject var context = GenerationContext()
@StateObject var generation = GenerationContext()
@State private var preparationPhase = "Downloading…"
@State private var downloadProgress: Double = 0
@ -37,7 +37,7 @@ struct LoadingView: View {
}
}
.animation(.easeIn, value: currentView)
.environmentObject(context)
.environmentObject(generation)
.onAppear {
Task.init {
let loader = PipelineLoader(model: model)
@ -59,7 +59,7 @@ struct LoadingView: View {
}
}
do {
context.pipeline = try await loader.prepare()
generation.pipeline = try await loader.prepare()
self.currentView = .textToImage
} catch {
self.currentView = .error("Could not load model, error: \(error)")

@ -94,18 +94,18 @@ struct ImageWithPlaceholder: View {
}
struct TextToImage: View {
@EnvironmentObject var context: GenerationContext
@EnvironmentObject var generation: GenerationContext
@State private var prompt = "Labrador in the style of Vermeer"
func submit() {
if case .running = context.state { return }
if case .running = generation.state { return }
Task {
context.state = .running(nil)
generation.state = .running(nil)
let interval: TimeInterval?
let image: CGImage?
(image, interval) = await context.generate(prompt: prompt, steps: steps, seed: seed) ?? (nil, nil)
context.state = .complete(prompt, image, interval)
(image, interval) = await generation.generate(prompt: prompt, steps: steps, seed: seed) ?? (nil, nil)
generation.state = .complete(prompt, image, interval)
}
}
@ -123,7 +123,7 @@ struct TextToImage: View {
.padding()
.buttonStyle(.borderedProminent)
}
ImageWithPlaceholder(state: $context.state)
ImageWithPlaceholder(state: $generation.state)
.scaledToFit()
Spacer()
}

Loading…
Cancel
Save