diff --git a/convert/convert.go b/convert/convert.go index 11d7e63a..6607c833 100644 --- 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) }