tests/models/jukebox/test_modeling_jukebox.py · 3a9476d1b412274bcc51143acaaee187e9d18120 · chenpangpang / transformers

"tests/models/roberta/test_modeling_roberta.py" did not exist on "345c23a60f86d7a786423c0e5274e5e58a20c7b2"

Arthur authored Dec 02, 2022

* fix cuda OOM by using single Prior

* only send to device when used

* use custom model

3a9476d1

test_modeling_jukebox.py 15.3 KB