runtime error

Exit code: 1. Reason: 2k [00:00<00:00, 4.35MB/s] vocab.json: 0%| | 0.00/2.78M [00:00<?, ?B/s] vocab.json: 100%|██████████| 2.78M/2.78M [00:00<00:00, 84.6MB/s] merges.txt: 0%| | 0.00/1.67M [00:00<?, ?B/s] merges.txt: 100%|██████████| 1.67M/1.67M [00:00<00:00, 55.5MB/s] tokenizer.json: 0%| | 0.00/7.03M [00:00<?, ?B/s] tokenizer.json: 100%|██████████| 7.03M/7.03M [00:00<00:00, 31.2MB/s] added_tokens.json: 0%| | 0.00/80.0 [00:00<?, ?B/s] added_tokens.json: 100%|██████████| 80.0/80.0 [00:00<00:00, 450kB/s] special_tokens_map.json: 0%| | 0.00/510 [00:00<?, ?B/s] special_tokens_map.json: 100%|██████████| 510/510 [00:00<00:00, 1.90MB/s] `torch_dtype` is deprecated! Use `dtype` instead! config.json: 0%| | 0.00/1.28k [00:00<?, ?B/s] config.json: 100%|██████████| 1.28k/1.28k [00:00<00:00, 4.47MB/s] Traceback (most recent call last): File "/app/app.py", line 20, in <module> model = LlavaQwen2ForCausalLM.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 288, in _wrapper return func(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 4905, in from_pretrained device_in_context = get_torch_context_manager_or_global_device() File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 300, in get_torch_context_manager_or_global_device device_in_context = torch.tensor([]).device File "/usr/local/lib/python3.10/site-packages/torch/utils/_device.py", line 103, in __torch_function__ return func(*args, **kwargs) 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: