Update LanguageModel.swift (#276)

Lakr233 · web-flow · commit cda4025b3857 · 2025-09-29T17:00:32.000+02:00
diff --git a/Sources/Models/LanguageModel.swift b/Sources/Models/LanguageModel.swift
@@ -472,7 +472,7 @@ public class LanguageModelWithStatefulKVCache: LanguageModel {
             Keys.inputIds: inputIds
         ]
         if isRequiringAttentionMask {
-            #if !((os(macOS) || (macCatalyst)) && arch(x86_64))
+            #if !((os(macOS) || targetEnvironment(macCatalyst)) && arch(x86_64))
             // TODO: Infer scalar type from cache or model I/O descriptors
             let attentionMask = MLTensor(zeros: [1, 1, 1, tokenCount + 1], scalarType: Float16.self)
             inputDictionary[Keys.attentionMask] = attentionMask

Original file line number	Diff line number	Diff line change
`@@ -472,7 +472,7 @@ public class LanguageModelWithStatefulKVCache: LanguageModel {`
`472`	`472`	`Keys.inputIds: inputIds`
`473`	`473`	`]`
`474`	`474`	`if isRequiringAttentionMask {`
`475`		`- #if !((os(macOS) \|\| (macCatalyst)) && arch(x86_64))`
	`475`	`+ #if !((os(macOS) \|\| targetEnvironment(macCatalyst)) && arch(x86_64))`
`476`	`476`	`// TODO: Infer scalar type from cache or model I/O descriptors`
`477`	`477`	`let attentionMask = MLTensor(zeros: [1, 1, 1, tokenCount + 1], scalarType: Float16.self)`
`478`	`478`	`inputDictionary[Keys.attentionMask] = attentionMask`