From 6694be5e5027b2f27f6eeeb51a5284a2b18129ad Mon Sep 17 00:00:00 2001 From: Michael Yang Date: Mon, 6 May 2024 14:00:50 -0700 Subject: [PATCH] convert/llama: use WriteSeeker --- convert/llama.go | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/convert/llama.go b/convert/llama.go index 6da46ec9..fb576e2e 100644 --- a/convert/llama.go +++ b/convert/llama.go @@ -5,7 +5,6 @@ import ( "fmt" "io" "log/slog" - "os" "regexp" "strings" @@ -159,11 +158,5 @@ func (m *LlamaModel) WriteGGUF(ws io.WriteSeeker) error { "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) }