tests/models/musicgen/test_modeling_musicgen.py · 869733ab621495b938d0754176f7f1e360ae7ea9 · chenpangpang / transformers · GitLab

Find file Blame History Permalink

[MusicGen] Fix integration tests (#25169) · 03f98f96
Sanchit Gandhi authored Jul 28, 2023
```
* move to device

* update with cuda values

* fix fp16

* more rigorous
```
03f98f96

test_modeling_musicgen.py 53 KB