Bläddra i källkod

seek to end of file when decoding older model formats

Jeffrey Morgan 1 år sedan
förälder
incheckning
d9a250e9b5
1 ändrade filer med 10 tillägg och 0 borttagningar
  1. 10 0
      llm/ggml.go

+ 10 - 0
llm/ggml.go

@@ -93,6 +93,8 @@ func (c *containerGGML) Name() string {
 }
 }
 
 
 func (c *containerGGML) Decode(ro *readSeekOffset) (model, error) {
 func (c *containerGGML) Decode(ro *readSeekOffset) (model, error) {
+	// file contents aren't decoded
+	ro.Seek(0, io.SeekEnd)
 	return nil, nil
 	return nil, nil
 }
 }
 
 
@@ -115,6 +117,10 @@ func (c *containerGGMF) Decode(ro *readSeekOffset) (model, error) {
 	}
 	}
 
 
 	c.version = version
 	c.version = version
+
+	// remaining file contents aren't decoded
+	ro.Seek(0, io.SeekEnd)
+
 	return nil, nil
 	return nil, nil
 }
 }
 
 
@@ -167,6 +173,10 @@ func (c *containerLORA) Decode(ro *readSeekOffset) (model, error) {
 	}
 	}
 
 
 	c.version = version
 	c.version = version
+
+	// remaining file contents aren't decoded
+	ro.Seek(0, io.SeekEnd)
+
 	return nil, nil
 	return nil, nil
 }
 }