Xuehai Pan
|
6bc61aa7af
|
Set TF32 flag for PyTorch cuDNN backend (#25075)
|
2023-07-25 08:04:48 -04:00 |
|
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 |
|