tests/models/owlvit/test_modeling_owlvit.py · a1d4563f7a2d78a5b29e4da46c76c90c4afe5331 · chenpangpang / transformers

Younes Belkada authored Nov 25, 2022

* `accelerate` support for `OwlViT`

- added `accelerate` support
- added slow `fp16` tests

* apply suggestions

a1d4563f

test_modeling_owlvit.py 30 KB