transformers/tests/models/seggpt
Raushan Turganbay 5ac264d8a8
Fix batching tests for new models (Mamba and SegGPT) (#29633)
* fix batchinng tests for new models

* Update tests/models/seggpt/test_modeling_seggpt.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

---------

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
2024-03-13 17:52:49 +00:00
..
__init__.py Adding SegGPT (#27735) 2024-02-26 18:17:19 +00:00
test_image_processing_seggpt.py Adding SegGPT (#27735) 2024-02-26 18:17:19 +00:00
test_modeling_seggpt.py Fix batching tests for new models (Mamba and SegGPT) (#29633) 2024-03-13 17:52:49 +00:00