jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rmv comments

Browse Source
This commit is contained in:
Josh Yan 2024-07-16 16:37:50 -07:00
parent 6ee22d5080
commit dea2204b82
1 changed files with 0 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
llm/ggml.go
View File

@ -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,