Update model_vision.go
This commit is contained in:
parent
359e1d5b19
commit
b3da8a319e
@ -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
|
totalSeqLen += seqLen
|
||||||
bounds = append(bounds, totalSeqLen*(m.spatialMergeSize*m.spatialMergeSize)+bounds[0])
|
bounds = append(bounds, totalSeqLen*(m.spatialMergeSize*m.spatialMergeSize)+bounds[0])
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user