tests/models/musicgen/test_modeling_musicgen.py · e367a9770fea240c001c958b7e0e39d4fa547519 · chenpangpang / transformers

Nripesh Niketan authored Jul 05, 2023

* Add mps function utils

* black formating

* format fix

* Added MPS functionality to transformers

* format fix

bd9dfc23

test_modeling_musicgen.py 52.4 KB