runtime error

Exit code: 1. Reason: noqa: E402 (after spaces) ^^^^^^^^^^^^ File "/app/model.py", line 20, in <module> pipeline = IdolPipeline.from_pretrained(os.environ.get('IDOL_WEIGHTS', 'speridlabs/idol'), device='cuda') ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/app/idol/pipeline.py", line 58, in from_pretrained return cls(snapshot_download(repo_or_dir, allow_patterns=WEIGHT_FILES), **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/app/idol/pipeline.py", line 42, in __init__ self.denoiser = load(LatentSeqMMFlowModel(**self.config['denoiser']['args']), 'denoiser') ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/app/idol/pipeline.py", line 40, in load return model.eval().requires_grad_(False).to(self.device) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1383, in to return self._apply(convert) ^^^^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.12/site-packages/torch/nn/modules/module.py", line 933, in _apply module._apply(fn) File "/usr/local/lib/python3.12/site-packages/torch/nn/modules/module.py", line 933, in _apply module._apply(fn) File "/usr/local/lib/python3.12/site-packages/torch/nn/modules/module.py", line 933, in _apply module._apply(fn) File "/usr/local/lib/python3.12/site-packages/torch/nn/modules/module.py", line 964, in _apply param_applied = fn(param) ^^^^^^^^^ File "/usr/local/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1369, in convert return t.to( ^^^^^ File "/usr/local/lib/python3.12/site-packages/torch/cuda/__init__.py", line 529, 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: