Hugging Face's logo

harmonai
/
maestro-150k

Diffusers
DanceDiffusionPipeline
audio-generation
Model card Files Community
3
patrickvonplaten commited on
Commit
271afe2
·
1 Parent(s): 9ac1f4d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -9,7 +9,7 @@ model_id = "harmonai/maestro-150k"
9
  pipeline = DiffusionPipeline.from_pretrained(model_id)
10
  pipeline = pipeline.to("cuda")
11
 
12
- audio = pipeline(audio_length_in_s=4.0).audios
13
 
14
  scipy.io.wavfile.write("maestro_test.wav", pipe.unet.sample_rate, audios)
15
  ```
@@ -28,7 +28,7 @@ model_id = "harmonai/maestro-150k"
28
  pipeline = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
29
  pipeline = pipeline.to("cuda")
30
 
31
- audio = pipeline(audio_length_in_s=4.0).audios
32
 
33
  scipy.io.wavfile.write("maestro_test.wav", pipe.unet.sample_rate, audios)
34
  ```
 
9
  pipeline = DiffusionPipeline.from_pretrained(model_id)
10
  pipeline = pipeline.to("cuda")
11
 
12
+ audios = pipeline(audio_length_in_s=4.0).audios
13
 
14
  scipy.io.wavfile.write("maestro_test.wav", pipe.unet.sample_rate, audios)
15
  ```
 
28
  pipeline = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
29
  pipeline = pipeline.to("cuda")
30
 
31
+ audios = pipeline(audio_length_in_s=4.0).audios
32
 
33
  scipy.io.wavfile.write("maestro_test.wav", pipe.unet.sample_rate, audios)
34
  ```