Arthur acc439ba17 Ci-jukebox (#20613) ...

* 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>

2022-12-06 16:14:03 +01:00

..

__init__.py

Add Jukebox model (replaces #16875) (#17826)

2022-11-10 21:05:27 +01:00

test_modeling_jukebox.py

Ci-jukebox (#20613)

2022-12-06 16:14:03 +01:00

test_tokenization_jukebox.py

Add Jukebox model (replaces #16875) (#17826)

2022-11-10 21:05:27 +01:00