mirror of
https://github.com/likelovewant/ollama-for-amd.git
synced 2025-12-24 07:28:27 +00: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