From 1ee01d6c50acfc5f10598805cebad7c90ded7bf0 Mon Sep 17 00:00:00 2001 From: ZachNagengast Date: Mon, 30 Dec 2024 22:13:08 -0800 Subject: [PATCH] Use canImport for MLTensor checks --- Sources/WhisperKit/Core/Text/TokenSampler.swift | 4 ++-- Sources/WhisperKit/Core/Utils/Utils.swift | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/Sources/WhisperKit/Core/Text/TokenSampler.swift b/Sources/WhisperKit/Core/Text/TokenSampler.swift index 4d833cf..2193161 100644 --- a/Sources/WhisperKit/Core/Text/TokenSampler.swift +++ b/Sources/WhisperKit/Core/Text/TokenSampler.swift @@ -28,7 +28,7 @@ open class GreedyTokenSampler: TokenSampling { self.decodingOptions = decodingOptions } - #if swift(>=5.10) + #if canImport(CoreML.MLState) @available(macOS 15, iOS 18, watchOS 11, visionOS 2, *) private func sampleWithMLTensor(logits: MLMultiArray) -> (token: Int, logprob: Float) { // Use MLTensor operations if available for sampling @@ -209,7 +209,7 @@ open class GreedyTokenSampler: TokenSampling { var completed = false var result: (token: Int, logprob: Float) - #if swift(>=5.10) + #if canImport(CoreML.MLState) if #available(macOS 15.0, iOS 18.0, watchOS 11.0, visionOS 2.0, *) { result = sampleWithMLTensor(logits: logits) } else { diff --git a/Sources/WhisperKit/Core/Utils/Utils.swift b/Sources/WhisperKit/Core/Utils/Utils.swift index 729a2a2..241997d 100644 --- a/Sources/WhisperKit/Core/Utils/Utils.swift +++ b/Sources/WhisperKit/Core/Utils/Utils.swift @@ -109,7 +109,7 @@ extension MLMultiArray { } } -#if swift(>=5.10) +#if canImport(CoreML.MLState) @available(macOS 15.0, iOS 18.0, watchOS 11.0, visionOS 2.0, *) public extension MLTensor { func asIntArray() -> [Int] {