runner.go: Check for zero length images
If we get a request with a zero length image, it will result in an out-of-bounds error when we pass the data to the image encoder.
This commit is contained in:
parent
771fab1dd8
commit
c2e8cbaa14
1 changed files with 4 additions and 0 deletions
|
@ -68,6 +68,10 @@ func (c *ImageContext) NewEmbed(llamaContext *llama.Context, data []byte, aspect
|
|||
return nil, nil
|
||||
}
|
||||
|
||||
if len(data) <= 0 {
|
||||
return nil, errors.New("received zero length image")
|
||||
}
|
||||
|
||||
hash := c.hashImage(data)
|
||||
|
||||
c.mu.Lock()
|
||||
|
|
Loading…
Reference in a new issue