runtime error

Exit code: 1. Reason: : 0%| | 0.00/3.75G [00:00<?, ?B/s] model.safetensors: 0%| | 10.5M/3.75G [00:01<11:05, 5.62MB/s] model.safetensors: 1%|▏ | 52.4M/3.75G [00:02<02:59, 20.6MB/s] model.safetensors: 3%|▎ | 115M/3.75G [00:03<01:40, 36.4MB/s]  model.safetensors: 27%|██▋ | 1.02G/3.75G [00:05<00:08, 310MB/s] model.safetensors: 35%|███▌ | 1.33G/3.75G [00:06<00:07, 309MB/s] model.safetensors: 44%|████▍ | 1.65G/3.75G [00:07<00:07, 299MB/s] model.safetensors: 60%|█████▉ | 2.23G/3.75G [00:08<00:03, 380MB/s] model.safetensors: 70%|██████▉ | 2.62G/3.75G [00:09<00:02, 378MB/s] model.safetensors: 88%|████████▊ | 3.30G/3.75G [00:10<00:00, 466MB/s] model.safetensors: 100%|█████████▉| 3.75G/3.75G [00:11<00:00, 327MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 127, in <module> model = AutoModel.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 559, in from_pretrained return model_class.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 3638, in from_pretrained state_dict = load_state_dict(resolved_archive_file) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 523, in load_state_dict return safe_load_file(checkpoint_file) File "/usr/local/lib/python3.10/site-packages/safetensors/torch.py", line 315, in load_file result[k] = f.get_tensor(k) File "/usr/local/lib/python3.10/site-packages/torch/utils/_device.py", line 79, in __torch_function__ return func(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 314, 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: