Arthur 3a9476d1b4 fix cuda OOM by using single Prior (#20486) ...

* fix cuda OOM by using single Prior

* only send to device when used

* use custom model

2022-12-02 09:05:45 +01:00

..

__init__.py

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

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

test_modeling_jukebox.py

fix cuda OOM by using single Prior (#20486)

2022-12-02 09:05:45 +01:00

test_tokenization_jukebox.py

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

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