|
@@ -317,12 +317,10 @@ func DecodeGGML(rs io.ReadSeeker, maxArraySize int) (*GGML, int64, error) {
|
|
maxArraySize = 1024
|
|
maxArraySize = 1024
|
|
}
|
|
}
|
|
|
|
|
|
- fmt.Println("errored 1")
|
|
|
|
rs = bufioutil.NewBufferedSeeker(rs, 32<<10)
|
|
rs = bufioutil.NewBufferedSeeker(rs, 32<<10)
|
|
|
|
|
|
var magic uint32
|
|
var magic uint32
|
|
if err := binary.Read(rs, binary.LittleEndian, &magic); err != nil {
|
|
if err := binary.Read(rs, binary.LittleEndian, &magic); err != nil {
|
|
- fmt.Println("errored 2")
|
|
|
|
return nil, 0, err
|
|
return nil, 0, err
|
|
}
|
|
}
|
|
|
|
|
|
@@ -337,24 +335,19 @@ func DecodeGGML(rs io.ReadSeeker, maxArraySize int) (*GGML, int64, error) {
|
|
case FILE_MAGIC_GGUF_BE:
|
|
case FILE_MAGIC_GGUF_BE:
|
|
c = &containerGGUF{ByteOrder: binary.BigEndian, maxArraySize: maxArraySize}
|
|
c = &containerGGUF{ByteOrder: binary.BigEndian, maxArraySize: maxArraySize}
|
|
default:
|
|
default:
|
|
- fmt.Println("errored 3")
|
|
|
|
return nil, 0, errors.New("invalid file magic")
|
|
return nil, 0, errors.New("invalid file magic")
|
|
}
|
|
}
|
|
|
|
|
|
- fmt.Println("valid magic")
|
|
|
|
model, err := c.Decode(rs)
|
|
model, err := c.Decode(rs)
|
|
if err != nil {
|
|
if err != nil {
|
|
return nil, 0, err
|
|
return nil, 0, err
|
|
}
|
|
}
|
|
|
|
|
|
- fmt.Println("valid decode")
|
|
|
|
offset, err := rs.Seek(0, io.SeekCurrent)
|
|
offset, err := rs.Seek(0, io.SeekCurrent)
|
|
if err != nil {
|
|
if err != nil {
|
|
- fmt.Println("invalid seek")
|
|
|
|
return nil, 0, err
|
|
return nil, 0, err
|
|
}
|
|
}
|
|
|
|
|
|
- fmt.Println("valid seek")
|
|
|
|
// final model type
|
|
// final model type
|
|
return &GGML{
|
|
return &GGML{
|
|
container: c,
|
|
container: c,
|