runtime error

Exit code: 1. Reason: /modeling_utils.py", line 4001, in from_pretrained model, missing_keys, unexpected_keys, mismatched_keys, offload_index, error_msgs = cls._load_pretrained_model( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 4141, in _load_pretrained_model convert_and_load_state_dict_in_model( File "/usr/local/lib/python3.10/site-packages/transformers/core_model_loading.py", line 879, in convert_and_load_state_dict_in_model realized_value, misc = mapping.convert( File "/usr/local/lib/python3.10/site-packages/transformers/core_model_loading.py", line 395, in convert futures if isinstance(futures[0], torch.Tensor) else [future.result() for future in futures] File "/usr/local/lib/python3.10/site-packages/transformers/core_model_loading.py", line 395, in <listcomp> futures if isinstance(futures[0], torch.Tensor) else [future.result() for future in futures] File "/usr/local/lib/python3.10/concurrent/futures/_base.py", line 451, in result return self.__get_result() File "/usr/local/lib/python3.10/concurrent/futures/_base.py", line 403, in __get_result raise self._exception File "/usr/local/lib/python3.10/concurrent/futures/thread.py", line 58, in run result = self.fn(*self.args, **self.kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/core_model_loading.py", line 497, in _job return _materialize_copy(tensor, device, dtype) File "/usr/local/lib/python3.10/site-packages/transformers/core_model_loading.py", line 491, in _materialize_copy tensor = tensor.to(device=device, dtype=dtype) File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 412, in _lazy_init torch._C._cuda_init() File "/usr/local/lib/python3.10/site-packages/spaces/zero/torch/patching.py", line 291, in _cuda_init_raise raise RuntimeError( RuntimeError: CUDA must not be initialized in the main process on Spaces with Stateless GPU environment. You can look at this Stacktrace to find out which part of your code triggered a CUDA init

Container logs: