transformers/tests/models/jukebox
Hz, Ji 50378cbf6c
device agnostic models testing (#27146)
* device agnostic models testing

* add decorator `require_torch_fp16`

* make style

* apply review suggestion

* Oops, the fp16 decorator was misused
2023-10-31 18:12:14 +01:00
..
__init__.py Add Jukebox model (replaces #16875) (#17826) 2022-11-10 21:05:27 +01:00
test_modeling_jukebox.py device agnostic models testing (#27146) 2023-10-31 18:12:14 +01:00
test_tokenization_jukebox.py Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00