Skip to content

Commit 9587712

Browse files
committed
Correctly show safety checker error.
1 parent 9c54eb4 commit 9587712

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

Diffusion-macOS/StatusView.swift

+1-1
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ struct StatusView: View {
7474
case .complete(_, let image, let lastSeed, let interval):
7575
guard let _ = image else {
7676
return HStack {
77-
Text("Safety checker triggered, please try a different prompt or seed")
77+
Text("Safety checker triggered, please try a different prompt or seed.")
7878
Spacer()
7979
}
8080
}

Diffusion/Pipeline/Pipeline.swift

+2-2
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,8 @@ class Pipeline {
6565
let interval = Date().timeIntervalSince(beginDate)
6666
print("Got images: \(images) in \(interval)")
6767

68-
// unwrap the 1 image we asked for
69-
guard let image = images.compactMap({ $0 }).first else { throw "Generation failed" }
68+
// Unwrap the 1 image we asked for, nil means safety checker triggered
69+
let image = images.compactMap({ $0 }).first
7070
return GenerationResult(image: image, lastSeed: theSeed, interval: interval)
7171
}
7272

0 commit comments

Comments
 (0)