riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity
b5be744d3c
transformers/tests/models/jukebox
History
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