Doesn't work on colab, no idea why
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.