runtime error

Exit code: 1. Reason: r on by calling `transformers.utils.move_cache()`. 0it [00:00, ?it/s] 0it [00:00, ?it/s] config_promax.json: 0%| | 0.00/1.26k [00:00<?, ?B/s] config_promax.json: 100%|██████████| 1.26k/1.26k [00:00<00:00, 5.34MB/s] (…)ffusion_pytorch_model_promax.safetensors: 0%| | 0.00/2.51G [00:00<?, ?B/s] (…)ffusion_pytorch_model_promax.safetensors: 1%| | 21.0M/2.51G [00:01<02:04, 20.0MB/s] (…)ffusion_pytorch_model_promax.safetensors: 2%|▏ | 52.4M/2.51G [00:02<02:11, 18.7MB/s] (…)ffusion_pytorch_model_promax.safetensors: 19%|█▉ | 472M/2.51G [00:03<00:12, 159MB/s]  (…)ffusion_pytorch_model_promax.safetensors: 45%|████▍ | 1.12G/2.51G [00:05<00:04, 310MB/s] (…)ffusion_pytorch_model_promax.safetensors: 58%|█████▊ | 1.47G/2.51G [00:06<00:03, 312MB/s] (…)ffusion_pytorch_model_promax.safetensors: 74%|███████▍ | 1.86G/2.51G [00:07<00:02, 323MB/s] (…)ffusion_pytorch_model_promax.safetensors: 100%|█████████▉| 2.51G/2.51G [00:08<00:00, 301MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 30, in <module> model.to(device="cuda", dtype=torch.float16) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1340, in to return self._apply(convert) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 900, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 927, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1326, in convert return t.to( File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 319, 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: