runtime error

Exit code: 1. Reason: | 33.2M/442M [00:01<00:12, 32.7MB/s] transformer_lora.bin: 100%|█████████▉| 442M/442M [00:01<00:00, 224MB/s] diffusion_pytorch_model.bin: 0%| | 0.00/1.01G [00:00<?, ?B/s] diffusion_pytorch_model.bin: 2%|▏ | 22.2M/1.01G [00:01<01:12, 13.6MB/s] diffusion_pytorch_model.bin: 27%|██▋ | 274M/1.01G [00:02<00:06, 111MB/s]  diffusion_pytorch_model.bin: 100%|█████████▉| 1.01G/1.01G [00:04<00:00, 247MB/s] transformer_lora.bin: 0%| | 0.00/442M [00:00<?, ?B/s] transformer_lora.bin: 10%|▉ | 43.7M/442M [00:01<00:09, 42.1MB/s] transformer_lora.bin: 100%|█████████▉| 442M/442M [00:01<00:00, 237MB/s] ./colorflow/models--TencentARC--ColorFlow/snapshots/0f80f02979276699eac09a427f3efa73150e7969 Traceback (most recent call last): File "/home/user/app/app.py", line 108, in <module> line_model.cuda() File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 905, in cuda return self._apply(lambda t: t.cuda(device)) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 797, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 797, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 797, in _apply module._apply(fn) [Previous line repeated 3 more times] File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 820, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 905, in <lambda> return self._apply(lambda t: t.cuda(device)) File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 247, 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: