kashif HF staff commited on
Commit
1704e81
1 Parent(s): fd6d2e9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -25,7 +25,7 @@ As depicted above the model takes as input a MIDI file and tokenizes it into a s
25
  ```python
26
  from diffusers import SpectrogramDiffusionPipeline, MidiProcessor
27
 
28
- pipe = SpectrogramDiffusionPipeline.from_pretrained("kashif/music-spectrogram-diffusion")
29
  pipe = pipe.to("cuda")
30
  processor = MidiProcessor()
31
 
 
25
  ```python
26
  from diffusers import SpectrogramDiffusionPipeline, MidiProcessor
27
 
28
+ pipe = SpectrogramDiffusionPipeline.from_pretrained("google/music-spectrogram-diffusion")
29
  pipe = pipe.to("cuda")
30
  processor = MidiProcessor()
31