runtime error
Exit code: 1. Reason: ", line 125, in <module> MODELS = {name: _load_to_cuda(name) for name in ALL_MODELS} File "/app/demos/musicgen_app.py", line 125, in <dictcomp> MODELS = {name: _load_to_cuda(name) for name in ALL_MODELS} File "/app/demos/musicgen_app.py", line 115, in _load_to_cuda m = MusicGen.get_pretrained(name, device='cuda') File "/app/audiocraft/models/musicgen.py", line 132, in get_pretrained lm = load_lm_model(name, device=device) File "/app/audiocraft/models/loaders.py", line 114, in load_lm_model model = builders.get_lm_model(cfg) File "/app/audiocraft/models/builders.py", line 97, in get_lm_model condition_provider = get_conditioner_provider(kwargs["dim"], cfg).to(cfg.device) File "/app/audiocraft/models/builders.py", line 141, in get_conditioner_provider conditioners[str(cond)] = ChromaStemConditioner( File "/app/audiocraft/modules/conditioners.py", line 544, in __init__ self.__dict__['demucs'] = pretrained.get_model('htdemucs').to(device) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1369, in to return self._apply(convert) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 928, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 928, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 928, in _apply module._apply(fn) [Previous line repeated 2 more times] File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 955, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1355, in convert return t.to( File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 412, in _lazy_init torch._C._cuda_init() RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx
Container logs:
Fetching error logs...