tests/models/seggpt/test_modeling_seggpt.py · 5ac264d8a8e014a7873c72fee91006d9e91a7bfd · chenpangpang / transformers

Fix batching tests for new models (Mamba and SegGPT) (#29633) · 5ac264d8

Raushan Turganbay authored Mar 13, 2024



* 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>

5ac264d8

test_modeling_seggpt.py 15.4 KB

Replace test_modeling_seggpt.py