From 18979ad4a1d40d04e3b981a477fa6323a40304b6 Mon Sep 17 00:00:00 2001
From: Michael Yang <mxyng@pm.me>
Date: Fri, 8 Mar 2024 15:40:16 -0800
Subject: [PATCH] convert: fix default shape

---
 convert/convert.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)
 		}