music-spectrogram-diffusion

kashif HF Staff commited on Mar 21, 2023

Commit

1704e81

1 Parent(s): fd6d2e9

Update README.md

Files changed (1) hide show

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ As depicted above the model takes as input a MIDI file and tokenizes it into a s
 ```python
 from diffusers import SpectrogramDiffusionPipeline, MidiProcessor
-pipe = SpectrogramDiffusionPipeline.from_pretrained("kashif/music-spectrogram-diffusion")
 pipe = pipe.to("cuda")
 processor = MidiProcessor()

 ```python
 from diffusers import SpectrogramDiffusionPipeline, MidiProcessor
+pipe = SpectrogramDiffusionPipeline.from_pretrained("google/music-spectrogram-diffusion")
 pipe = pipe.to("cuda")
 processor = MidiProcessor()