@@ -4,6 +4,7 @@ import Testing
44@testable import AnyLanguageModel
55
66#if MLX
7+ import MLXLMCommon
78 private let shouldRunMLXTests = {
89 // Enable when explicitly requested via environment variable
910 if ProcessInfo . processInfo. environment [ " ENABLE_MLX_TESTS " ] != nil {
@@ -154,7 +155,11 @@ import Testing
154155 )
155156 ] )
156157 let session = LanguageModelSession ( model: visionModel, transcript: transcript)
157- let response = try await session. respond ( to: " " )
158+ var options = GenerationOptions ( )
159+ var mlxOptions = MLXLanguageModel . CustomGenerationOptions. default
160+ mlxOptions. userInputProcessing = . resize( to: CGSize ( width: 512 , height: 512 ) )
161+ options [ custom: MLXLanguageModel . self] = mlxOptions
162+ let response = try await session. respond ( to: " " , options: options)
158163 #expect( !response. content. isEmpty)
159164 }
160165
@@ -168,7 +173,11 @@ import Testing
168173 )
169174 ] )
170175 let session = LanguageModelSession ( model: visionModel, transcript: transcript)
171- let response = try await session. respond ( to: " " )
176+ var options = GenerationOptions ( )
177+ var mlxOptions = MLXLanguageModel . CustomGenerationOptions. default
178+ mlxOptions. userInputProcessing = . resize( to: CGSize ( width: 512 , height: 512 ) )
179+ options [ custom: MLXLanguageModel . self] = mlxOptions
180+ let response = try await session. respond ( to: " " , options: options)
172181 #expect( !response. content. isEmpty)
173182 }
174183
0 commit comments