transformers
756820c0 - [emu3] Fix OOM in test_model_generation_multi_image and update expected strings for A10G

Commit
10 days ago
[emu3] Fix OOM in test_model_generation_multi_image and update expected strings for A10G - Reduce image processor resolution to 32x32 tokens (256x256 pixels) in test_model_generation_multi_image to avoid SDPA OOM on A10G (two 64x64 images produce ~8192 tokens, requiring 8+ GiB for attention alongside the 4-bit quantized model) - Update ("cuda", 8) expected strings for both test_model_generation_batched and test_model_generation_multi_image to match actual A10G outputs - Use 200x200 image downloads consistently across all integration tests Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
Author
Parents
Loading