transformers/tests/models/aria
Yao Matrix 3bd1c20149
enable misc cases on XPU & use device agnostic APIs for cases in tests (#38192)
* use device agnostic APIs in tests

Signed-off-by: Matrix Yao <matrix.yao@intel.com>

* more

Signed-off-by: Matrix Yao <matrix.yao@intel.com>

* fix style

Signed-off-by: Matrix Yao <matrix.yao@intel.com>

* add reset_peak_memory_stats API

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* update

---------

Signed-off-by: Matrix Yao <matrix.yao@intel.com>
Signed-off-by: YAO Matrix <matrix.yao@intel.com>
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2025-05-20 10:09:01 +02:00
..
__init__.py Add Aria (#34157) 2024-12-06 12:17:34 +01:00
test_image_processing_aria.py [llava] one pixel is missing from padding when length is odd (#37819) 2025-05-06 13:11:26 +02:00
test_modeling_aria.py enable misc cases on XPU & use device agnostic APIs for cases in tests (#38192) 2025-05-20 10:09:01 +02:00
test_processor_aria.py [vlm] adjust max length for special tokens (#37342) 2025-04-16 20:49:20 +02:00