We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent cab24e9 commit 4361735Copy full SHA for 4361735
swift/StableDiffusion/pipeline/TextEncoderT5.swift
@@ -91,9 +91,9 @@ public struct TextEncoderT5: TextEncoderT5Model {
91
92
let attentionMaskName = "attention_mask"
93
var attentionMask: [Int] = inputIds.map { token in
94
- token == padToken ? maskToken : padToken
+ token == padToken ? padToken : maskToken
95
}
96
- attentionMask[0] = bosToken
+ attentionMask[0] = eosToken
97
98
let floatIds = inputIds.map { Float32($0) }
99
let floatMask = attentionMask.map { Float32($0) }
0 commit comments