Commit b19fe2c
committed
mtmd: fix Pixtral OOM with large images by capping image_size to 1024
Mistral Small 2506 models using Pixtral vision encoder were running out
of GPU memory when processing images larger than 1024x1024 pixels due to
exponential memory growth from unlimited image size.
This fix applies the same 1024x1024 limit used by Qwen2VL models to
prevent OOM issues while maintaining compatibility with existing models.
Fixes #143101 parent 40bfa04 commit b19fe2c
1 file changed
+3
-0
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
2211 | 2211 | | |
2212 | 2212 | | |
2213 | 2213 | | |
| 2214 | + | |
| 2215 | + | |
| 2216 | + | |
2214 | 2217 | | |
2215 | 2218 | | |
2216 | 2219 | | |
| |||
0 commit comments