tests/models/jukebox/test_modeling_jukebox.py · acc439ba172027152fb394c28b6f55a8df785326 · chenpangpang / transformers

Ci-jukebox (#20613) · acc439ba

Arthur authored Dec 06, 2022



* fix cuda OOM by using single Prior

* only send to device when used

* use custom model

* Skip the big slow test

* Update tests/models/jukebox/test_modeling_jukebox.py
Co-authored-by: Yih-Dar <2521628+ydshieh@users.noreply.github.com>
Co-authored-by: Yih-Dar <2521628+ydshieh@users.noreply.github.com>

acc439ba

test_modeling_jukebox.py 15.4 KB

Replace test_modeling_jukebox.py