Sfoglia il codice sorgente

convert/llama: use WriteSeeker

Michael Yang 1 anno fa
parent
commit
6694be5e50
1 ha cambiato i file con 1 aggiunte e 8 eliminazioni
  1. 1 8
      convert/llama.go

+ 1 - 8
convert/llama.go

@@ -5,7 +5,6 @@ import (
 	"fmt"
 	"fmt"
 	"io"
 	"io"
 	"log/slog"
 	"log/slog"
-	"os"
 	"regexp"
 	"regexp"
 	"strings"
 	"strings"
 
 
@@ -159,11 +158,5 @@ func (m *LlamaModel) WriteGGUF(ws io.WriteSeeker) error {
 		"tokenizer.ggml.add_eos_token":    false,
 		"tokenizer.ggml.add_eos_token":    false,
 	}
 	}
 
 
-	f, err := os.CreateTemp("", "ollama-gguf")
-	if err != nil {
-		return err
-	}
-	defer f.Close()
-
-	return llm.NewGGUFV3(m.Params.ByteOrder).Encode(f, kv, m.Tensors)
+	return llm.NewGGUFV3(m.Params.ByteOrder).Encode(ws, kv, m.Tensors)
 }
 }