Feature request: Fast image processor for Fuyu
Feature request: Fast image processor for Fuyu: a task in LegoFlow-SWE (Harbor dataset). The Fuyu model currently only has a slow image processor ( FuyuImageProcessor ) that relies on PIL/numpy for image transformations. This is a bottleneck during inference, especially when processing large…
The task
The Fuyu model currently only has a slow image processor (`FuyuImageProcessor`) that relies on PIL/numpy for image transformations. This is a bottleneck during inference, especially when processing large batches or high-resolution images, because operations happen on the CPU and cannot leverage GPU acceleration.
Part of Lego-X/LegoFlow-SWE.