Doesn't work on colab, no idea why

#2
by Nexyboye - opened

Hi, I just want to try the model out on colab, this is what I've got so far:

!pip install diffusers
!pip install note-seq
!pip install onnx
!pip install onnxruntime

from diffusers import SpectrogramDiffusionPipeline, MidiProcessor

pipe = SpectrogramDiffusionPipeline.from_pretrained("google/music-spectrogram-diffusion") # <---------- The error is here
pipe = pipe.to("cuda")
processor = MidiProcessor()

and this is the error:
NoSuchFile: [ONNXRuntimeError] : 3 : NO_SUCHFILE : Load model from /root/.cache/huggingface/hub/models--google--music-spectrogram-diffusion/snapshots/e2e5316f00d986686a9d09380808c0b1511b9820/model.onnx failed:Load model /root/.cache/huggingface/hub/models--google--music-spectrogram-diffusion/snapshots/e2e5316f00d986686a9d09380808c0b1511b9820/model.onnx failed. File doesn't exist

I'm running it on a T4 gpu, and also tried restarting runtime a few times, but no luck.

Sign up or log in to comment