| import audiofile | |
| import numpy as np | |
| from audiocraft import AudioGen | |
| text_list = ['dogs barging in the street', | |
| 'music'] | |
| sound_generator = AudioGen(duration=.74, | |
| device='cuda:0').to('cuda:0').eval() | |
| x = sound_generator.generate(text_list) # [bs, 1, 7680] | |
| # print('demo', x.shape) | |
| x = x[0, :, :].detach().cpu().numpy() | |
| x /= np.abs(x).max() + 1e-7 | |
| audiofile.write('del_seane.wav', x, 16000) | |