Update model_vision.go

This commit is contained in:
Bruce MacDonald 2025-05-01 11:06:06 -07:00
parent 359e1d5b19
commit b3da8a319e

View File

@ -281,7 +281,6 @@ func (m *VisionModel) windowIndex(ctx ml.Context, grid *Grid) (ml.Tensor, []int)
}
}
// Update total sequence length and append to cuWindowSeqlens
totalSeqLen += seqLen
bounds = append(bounds, totalSeqLen*(m.spatialMergeSize*m.spatialMergeSize)+bounds[0])
}