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
|
return nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if len(data) <= 0 {
|
||||||
|
return nil, errors.New("received zero length image")
|
||||||
|
}
|
||||||
|
|
||||||
hash := c.hashImage(data)
|
hash := c.hashImage(data)
|
||||||
|
|
||||||
c.mu.Lock()
|
c.mu.Lock()
|
||||||
|
|
Loading…
Reference in a new issue