Skip to content

Commit f83d531

Browse files
committed
Remove type casting for pixel values (mlx-vlm #398)
1 parent 410c89c commit f83d531

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
@@ -1778,10 +1778,8 @@ public class Gemma3n: Module, VLMModel, KVCacheDimensionProvider {
17781778
}
17791779

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

0 commit comments

Comments
 (0)