runtime error

Exit code: 1. Reason: -op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?) warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta ' Done Init Perceive Sampler ... Done Traceback (most recent call last): File "/home/user/app/app.py", line 9, in <module> model = AutoModelForCausalLM.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 511, in from_pretrained return model_class.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 2876, in from_pretrained model = cls(config, *model_args, **model_kwargs) File "/home/user/.cache/huggingface/modules/transformers_modules/Lin-Chen/ShareCaptioner/a2679a4033bf648394181e8a118b813fd5d34b7a/modeling_InternLM_XComposer.py", line 70, in __init__ self.internlm_model.to_empty(device=config.device).to(torch.float16) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1039, in to_empty return self._apply(lambda t: torch.empty_like(t, device=device), recurse=recurse) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 810, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 810, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 833, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1039, in <lambda> return self._apply(lambda t: torch.empty_like(t, device=device), recurse=recurse) File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 298, 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 Init InternLM ...

Container logs: