tcsenpai/ollama
1
0
Fork 0
mirror of https://github.com/tcsenpai/ollama.git synced 2025-06-07 11:45:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

convert: fix shape

Browse Source
This commit is contained in:
Michael Yang 2024-03-10 10:41:40 -07:00
parent ac64cd4ef9
commit 9ea492f1ce
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
convert/convert.go
View File

@ -103,9 +103,9 @@ func ReadSafeTensors(fn string, offset uint64) ([]llm.Tensor, uint64, error) {
return []llm.Tensor{}, 0, err return []llm.Tensor{}, 0, err
} }
shape := [4]uint64{1, 1, 1, 1} shape := []uint64{0, 0, 0, 0}
for cnt, s := range data.Shape { for i := range data.Shape {
shape[cnt] = uint64(s) shape[i] = uint64(data.Shape[i])
} }
t := llm.Tensor{ t := llm.Tensor{