mirrors/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-05-11 18:36:41 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix image processing

Browse source 
python built-in `round()` rounds to the nearest even number if the value
is in the middle

https://docs.python.org/3/library/functions.html#round
This commit is contained in:
Michael Yang 2025-04-29 11:31:49 -07:00 • committed by Bruce MacDonald
parent d2cfdbc1bc
commit 468efd29df
1 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
model/models/qwen25vl/process_image.go
View file

@ -50,14 +50,12 @@ func (p *ImageProcessor) SmartResize(height, width int) (int, int) {
if height < factor || width < factor {
panic(fmt.Sprintf("height:%d or width:%d must be larger than factor:%d", height, width, factor))
} else if float64(max(height, width))/float64(min(height, width)) > 200 {
aspectRatio := float64(max(height, width)) / float64(min(height, width))
panic(fmt.Sprintf("absolute aspect ratio must be smaller than 200, got %f", aspectRatio))
} else if aspectRatio := max(height, width) / min(height, width); aspectRatio > 200 {
panic(fmt.Sprintf("absolute aspect ratio must be smaller than 200, got %v", aspectRatio))
}
round := func(x float64) int {
return int(math.Round(x))
}
round := func(x float64) int { return int(math.RoundToEven(x)) }
hBar := round(float64(height)/float64(factor)) * factor
wBar := round(float64(width)/float64(factor)) * factor