Skip to content

GitLab

Commit 9ea492f1 authored by Michael Yang's avatar Michael Yang
Browse files

convert: fix shape

parent ac64cd4e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
convert/convert.go
View file @ 9ea492f1
...@@ -103,9 +103,9 @@ func ReadSafeTensors(fn string, offset uint64) ([]llm.Tensor, uint64, error) { ...@@ -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{
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment