浏览代码

close open files

Michael Yang 1 年之前
父节点
当前提交
e26085b921
共有 3 个文件被更改,包括 3 次插入0 次删除
  1. 1 0
      llm/llama_darwin.go
  2. 1 0
      llm/llm.go
  3. 1 0
      server/images.go

+ 1 - 0
llm/llama_darwin.go

@@ -39,6 +39,7 @@ func initBackend() error {
 		if err != nil {
 			return err
 		}
+		defer actual.Close()
 
 		actualSum := sha256.New()
 		if _, err := io.Copy(actualSum, actual); err != nil {

+ 1 - 0
llm/llm.go

@@ -28,6 +28,7 @@ func New(model string, adapters []string, opts api.Options) (LLM, error) {
 	if err != nil {
 		return nil, err
 	}
+	defer f.Close()
 
 	ggml, err := DecodeGGML(f, ModelFamilyLlama)
 	if err != nil {

+ 1 - 0
server/images.go

@@ -1117,6 +1117,7 @@ func uploadBlobChunked(ctx context.Context, mp ModelPath, url string, layer *Lay
 	if err != nil {
 		return err
 	}
+	defer f.Close()
 
 	totalUploaded := 0