convert: fix default shape

18979ad4 · Michael Yang · 8e0ef931 · 18979ad4
Commit 18979ad4 authored Mar 08, 2024 by Michael Yang
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

convert/convert.go convert/convert.go +1 -1

No files found.
--- a/convert/convert.go
+++ b/convert/convert.go
@@ -103,7 +103,7 @@ func ReadSafeTensors(fn string, offset uint64) ([]llm.Tensor, uint64, error) {
 			return []llm.Tensor{}, 0, err
 		}

-		shape := [4]uint64{0, 0, 0, 0}
+		shape := [4]uint64{1, 1, 1, 1}
 		for cnt, s := range data.Shape {
 			shape[cnt] = uint64(s)
 		}