@@ -100,12 +100,12 @@ public static NDList decode(NDManager manager, byte[] byteArray) {
100
100
try {
101
101
if (byteArray [0 ] == 'P' && byteArray [1 ] == 'K' ) {
102
102
return decodeNumpy (manager , new ByteArrayInputStream (byteArray ));
103
- } else if (byteArray [0 ] == (byte ) 0x39
103
+ } else if (byteArray [0 ] == (byte ) 0x93
104
104
&& byteArray [1 ] == 'N'
105
105
&& byteArray [2 ] == 'U'
106
106
&& byteArray [3 ] == 'M' ) {
107
107
return new NDList (
108
- NDSerializer .decode (manager , new ByteArrayInputStream (byteArray )));
108
+ NDSerializer .decodeNumpy (manager , new ByteArrayInputStream (byteArray )));
109
109
} else if (byteArray [8 ] == '{' ) {
110
110
return decodeSafetensors (manager , new ByteArrayInputStream (byteArray ));
111
111
}
@@ -144,11 +144,11 @@ public static NDList decode(NDManager manager, InputStream is) {
144
144
if (magic [0 ] == 'P' && magic [1 ] == 'K' ) {
145
145
// assume this is npz file
146
146
return decodeNumpy (manager , pis );
147
- } else if (magic [0 ] == (byte ) 0x39
147
+ } else if (magic [0 ] == (byte ) 0x93
148
148
&& magic [1 ] == 'N'
149
149
&& magic [2 ] == 'U'
150
150
&& magic [3 ] == 'M' ) {
151
- return new NDList (NDSerializer .decode (manager , pis ));
151
+ return new NDList (NDSerializer .decodeNumpy (manager , pis ));
152
152
} else if (magic [8 ] == '{' ) {
153
153
return decodeSafetensors (manager , pis );
154
154
}
0 commit comments