Fix invalid mask implementation

apple · Sep 21, 2024 · b149946 · b149946
1 parent cab24e9
commit b149946
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/swift/StableDiffusion/pipeline/TextEncoderT5.swift b/swift/StableDiffusion/pipeline/TextEncoderT5.swift
@@ -91,7 +91,7 @@ public struct TextEncoderT5: TextEncoderT5Model {
 
         let attentionMaskName = "attention_mask"
         var attentionMask: [Int] = inputIds.map { token in
-            token == padToken ? maskToken : padToken
+            token == padToken ? padToken : maskToken
         }
         attentionMask[0] = bosToken