transformers/tests/dit
NielsRogge 03c14a515f
[Tests] Fix DiT test (#16218)
* Fix device

* Clean up

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
2022-03-17 10:53:57 +01:00
..
__init__.py Add Document Image Transformer (DiT) (#15984) 2022-03-10 11:34:44 +01:00
test_modeling_dit.py [Tests] Fix DiT test (#16218) 2022-03-17 10:53:57 +01:00