Skip to content

Commit 0ea32f6

Browse files
committed
Fix iOS compilation.
I changed the return type of the generation pipeline but did not update the iOS code. Addresses huggingface#26.
1 parent c456eb8 commit 0ea32f6

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

Diffusion/Views/TextToImage.swift

+9-5
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ struct ImageWithPlaceholder: View {
6666
let fraction = Double(step) / Double(progress.stepCount)
6767
let label = "Step \(step) of \(progress.stepCount)"
6868
return AnyView(ProgressView(label, value: fraction, total: 1).padding())
69-
case .complete(let lastPrompt, let image, let interval):
69+
case .complete(let lastPrompt, let image, _, let interval):
7070
guard let theImage = image else {
7171
return AnyView(Image(systemName: "exclamationmark.triangle").resizable())
7272
}
@@ -86,6 +86,8 @@ struct ImageWithPlaceholder: View {
8686
)
8787
}.frame(maxHeight: 25)
8888
})
89+
case .failed(_):
90+
return AnyView(Image(systemName: "exclamationmark.triangle").resizable())
8991
}
9092
}
9193
}
@@ -97,10 +99,12 @@ struct TextToImage: View {
9799
if case .running = generation.state { return }
98100
Task {
99101
generation.state = .running(nil)
100-
let interval: TimeInterval?
101-
let image: CGImage?
102-
let result = await generation.generate()
103-
generation.state = .complete(generation.positivePrompt, image, interval)
102+
do {
103+
let result = try await generation.generate()
104+
generation.state = .complete(generation.positivePrompt, result.image, result.lastSeed, result.interval)
105+
} catch {
106+
generation.state = .failed(error)
107+
}
104108
}
105109
}
106110

0 commit comments

Comments
 (0)