浏览代码

convert: fix default shape

Michael Yang 1 年之前
父节点
当前提交
18979ad4a1
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      convert/convert.go

+ 1 - 1
convert/convert.go

@@ -103,7 +103,7 @@ func ReadSafeTensors(fn string, offset uint64) ([]llm.Tensor, uint64, error) {
 			return []llm.Tensor{}, 0, err
 			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 {
 		for cnt, s := range data.Shape {
 			shape[cnt] = uint64(s)
 			shape[cnt] = uint64(s)
 		}
 		}