matt/ollama37
1
0
Fork 0
mirror of https://github.com/dogkeeper886/ollama37.git synced 2025-12-11 16:26:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

use gzip for runner embedding (#2067)

Browse Source
This commit is contained in:
Jeffrey Morgan
2024-01-19 13:23:03 -05:00
committed by GitHub
parent 62976087c6
commit dc88cc3981
4 changed files with 18 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
llm/payload_common.go
View File

@@ -1,7 +1,7 @@
package llm
import (
"compress/bzip2"
"compress/gzip"
"errors"
"fmt"
"io"
@@ -182,9 +182,12 @@ func extractDynamicLibs(workDir, glob string) ([]string, error) {
}
src := io.Reader(srcFile)
filename := file
if strings.HasSuffix(file, ".bz2") {
src = bzip2.NewReader(src)
filename = strings.TrimSuffix(filename, ".bz2")
if strings.HasSuffix(file, ".gz") {
src, err = gzip.NewReader(src)
if err != nil {
return fmt.Errorf("decompress payload %s: %v", file, err)
}
filename = strings.TrimSuffix(filename, ".gz")
}
destFile := filepath.Join(targetDir, filepath.Base(filename))
@@ -229,9 +232,12 @@ func extractPayloadFiles(workDir, glob string) error {
}
src := io.Reader(srcFile)
filename := file
if strings.HasSuffix(file, ".bz2") {
src = bzip2.NewReader(src)
filename = strings.TrimSuffix(filename, ".bz2")
if strings.HasSuffix(file, ".gz") {
src, err = gzip.NewReader(src)
if err != nil {
return fmt.Errorf("decompress payload %s: %v", file, err)
}
filename = strings.TrimSuffix(filename, ".gz")
}
destFile := filepath.Join(workDir, filepath.Base(filename))