mirror of
https://github.com/ollama/ollama.git
synced 2026-03-11 17:34:04 -05:00
fix test
This commit is contained in:
committed by
Michael Yang
parent
207332078f
commit
e0cd511661
@@ -35,7 +35,7 @@ func splitDim(t Tensor, dim int, splits ...split) iter.Seq[*ggml.Tensor] {
|
||||
|
||||
slice := split.slices
|
||||
if len(slice) == 0 {
|
||||
slice := slices.Repeat([]tensor.Slice{nil}, len(shape))
|
||||
slice = slices.Repeat([]tensor.Slice{nil}, len(shape))
|
||||
slice[dim] = tensor.S(offset, offset+int(shape[dim]))
|
||||
offset += int(shape[dim])
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user