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:
Jesse Gross 2024-11-06 13:14:18 -08:00 committed by Jesse Gross
parent 771fab1dd8
commit c2e8cbaa14

View file

@ -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()