chaowenguo commited on
Commit
a2823ab
·
verified ·
1 Parent(s): 964db7e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -12,7 +12,7 @@ model = transformers.MusicgenMelodyForConditionalGeneration.from_pretrained('cha
12
 
13
  result = []
14
  for _ in builtins.range(9):
15
- inputs = processor(audio=result[-1] if result else None, sampling_rate=model.config.audio_encoder.sampling_rate, text='romance piano and violin background music for dancing', padding=True, return_tensors='pt').to('cuda')
16
  audio_values = model.generate(**inputs, do_sample=True, guidance_scale=3, max_new_tokens=1000)
17
  result += audio_values[0, 0].cpu().numpy(),
18
 
 
12
 
13
  result = []
14
  for _ in builtins.range(9):
15
+ inputs = processor(audio=result[-1] if result else None, sampling_rate=model.config.audio_encoder.sampling_rate, text='loud romance piano and violin background music for dancing', padding=True, return_tensors='pt').to('cuda')
16
  audio_values = model.generate(**inputs, do_sample=True, guidance_scale=3, max_new_tokens=1000)
17
  result += audio_values[0, 0].cpu().numpy(),
18