transformers/tests/models/musicgen
Sanchit Gandhi 03f98f9683
[MusicGen] Fix integration tests (#25169)
* move to device

* update with cuda values

* fix fp16

* more rigorous
2023-07-28 18:50:15 +01:00
..
__init__.py Add Musicgen (#24109) 2023-06-29 14:48:59 +01:00
test_modeling_musicgen.py [MusicGen] Fix integration tests (#25169) 2023-07-28 18:50:15 +01:00
test_processing_musicgen.py Add Musicgen (#24109) 2023-06-29 14:48:59 +01:00