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