pali112
/
ollama
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix test

This commit is contained in:
Michael Yang 2025-10-07 16:46:37 -07:00 committed by Michael Yang
parent 207332078f
commit e0cd511661
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
convert/tensor.go
View File

@ -35,7 +35,7 @@ func splitDim(t Tensor, dim int, splits ...split) iter.Seq[*ggml.Tensor] {
slice := split.slices slice := split.slices
if len(slice) == 0 { 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])) slice[dim] = tensor.S(offset, offset+int(shape[dim]))
offset += int(shape[dim]) offset += int(shape[dim])
} }