From 29ecfe493b99b270a2971ae1e3403b2f6b330eb5 Mon Sep 17 00:00:00 2001 From: Josh Yan Date: Thu, 11 Jul 2024 17:56:51 -0700 Subject: [PATCH] write --- llm/gguf.go | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/llm/gguf.go b/llm/gguf.go index 4d343a1bd..e35222cf4 100644 --- a/llm/gguf.go +++ b/llm/gguf.go @@ -695,3 +695,21 @@ func (llm *gguf) Encode(ws io.WriteSeeker, kv KV, tensors []Tensor) error { func (gguf) padding(offset, align int64) int64 { return (align - offset%align) % align } + +// Reader and WriterTo +type GGUFWriter struct { + KV KV + T []*Tensor +} + +var _ io.Reader = (*GGUFWriter)(nil) + +var _ io.WriterTo = (*GGUFWriter)(nil) + +func (GGUFWriter) Read([]byte) (int, error) { + panic("not implemeneted") +} + +func (gguf GGUFWriter) WriteTo(w io.Writer) (int64, error) { + +}