5 Commits

Author	SHA1	Message	Date
Yih-Dar	5b28b78332	Update Jukebox tests (#21984) ... * update expected values for jukebox * update expected values for jukebox * update expected values for jukebox * update expected values for jukebox * update expected values for jukebox --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>	2023-03-07 04:20:14 +01:00
Aaron Gokaslan	5e8c8eb5ba	Apply ruff flake8-comprehensions (#21694)	2023-02-22 09:14:54 +01:00
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
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
Arthur	61a51f5f23	Add Jukebox model (replaces #16875) (#17826)	2022-11-10 21:05:27 +01:00