Skip to content

Commit cf7f3f2

Browse files
committed
Remove type casting for pixel values (mlx-vlm #398)
1 parent 8e1d7d5 commit cf7f3f2

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

Libraries/MLXVLM/Models/Gemma3n.swift

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1777,10 +1777,8 @@ public class Gemma3n: Module, VLMModel, KVCacheDimensionProvider {
17771777
}
17781778

17791779
// Process vision features
1780-
if let pixelValues = pixelValues {
1781-
let pixelValuesTyped = pixelValues.asType(languageModel.model.embedTokens.weight.dtype)
1782-
let imageFeatures = getImageFeatures(pixelValuesTyped)
1783-
1780+
if let pixelValues {
1781+
let imageFeatures = getImageFeatures(pixelValues)
17841782
return mergeMultimodalAndText(
17851783
inputIds: inputIds,
17861784
inputsEmbeds: inputsEmbeds,

0 commit comments

Comments
 (0)