Fix invalid mask implementation

KyuHwan00 · KyuHwan00 · commit b149946b85b4 · 2024-09-21T14:43:19.000+09:00
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
 

Original file line number	Diff line number	Diff line change
`@@ -91,7 +91,7 @@ public struct TextEncoderT5: TextEncoderT5Model {`
`91`	`91`
`92`	`92`	`let attentionMaskName = "attention_mask"`
`93`	`93`	`var attentionMask: [Int] = inputIds.map { token in`
`94`		`- token == padToken ? maskToken : padToken`
	`94`	`+ token == padToken ? padToken : maskToken`
`95`	`95`	`}`
`96`	`96`	`attentionMask[0] = bosToken`
`97`	`97`