runtime error

Exit code: 1. Reason: ��| 4/4 [01:12<00:00, 18.15s/steps] Downloading shards: 100%|██████████| 4/4 [01:12<00:00, 18.11s/steps] /usr/local/lib/python3.10/site-packages/torch/nn/modules/transformer.py:286: UserWarning: enable_nested_tensor is True, but self.use_nested_tensor is False because encoder_layer.norm_first was True warnings.warn(f"enable_nested_tensor is True, but self.use_nested_tensor is False because {why_not_sparsity_fast_path}") config.json: 0%| | 0.00/1.99k [00:00<?, ?B/s] config.json: 100%|██████████| 1.99k/1.99k [00:00<00:00, 13.5MB/s] Loading checkpoint shards: 0%| | 0/4 [00:00<?, ?steps/s] Loading checkpoint shards: 0%| | 0/4 [00:00<?, ?steps/s] Traceback (most recent call last): File "/home/user/app/app.py", line 7, in <module> model, audio_processor, tokenizer = load_model("FreedomIntelligence/Soundwave", device) File "/home/user/app/Soundwave/run_inference.py", line 19, in load_model model = SoundwaveForCausalLM.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 3903, in from_pretrained ) = cls._load_pretrained_model( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 4377, in _load_pretrained_model new_error_msgs, offload_index, state_dict_index = _load_state_dict_into_meta_model( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 933, in _load_state_dict_into_meta_model set_module_tensor_to_device(model, param_name, param_device, **set_module_kwargs) File "/usr/local/lib/python3.10/site-packages/accelerate/utils/modeling.py", line 416, in set_module_tensor_to_device new_value = value.to(device) File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 302, 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: