File size: 422 Bytes
fe62fb4
 
54adc39
dcfe0d4
c5e1f80
dcfe0d4
54adc39
dcfe0d4
 
 
c5e1f80
 
e366cd5
1
2
3
4
5
6
7
8
9
10
11
12
13
14
import audiofile
import numpy as np
from audiocraft import AudioGen
text_list = ['dogs barging in the street', 'people po']

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)