mirror of
https://github.com/dogkeeper886/ollama37.git
synced 2025-12-10 15:57:04 +00:00
chore: update mllama to use ollama engine (#10637)
This commit is contained in:
@@ -208,7 +208,7 @@ func (m *VisionModel) Forward(ctx ml.Context, pixelValues ml.Tensor) ml.Tensor {
|
||||
}
|
||||
|
||||
hiddenStates = m.LayerNormPost.Forward(ctx, hiddenStates, m.eps)
|
||||
hiddenStates = hiddenStates.Unpad(ctx, 0, 1, 0, 0)
|
||||
hiddenStates = hiddenStates.Pad(ctx, 0, -1, 0, 0)
|
||||
hiddenStates = m.VisionAdapter.Forward(ctx, hiddenStates, m.VisionOptions)
|
||||
return hiddenStates
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user