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