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