runtime error

Exit code: 1. Reason: #auto-classes - If you are the owner of the model architecture code, please modify your model class such that it inherits from `GenerationMixin` (after `PreTrainedModel`, otherwise you'll get an exception). - If you are not the owner of the model architecture class, please contact the model code owner to update it. Traceback (most recent call last): File "/home/user/app/app.py", line 18, in <module> model = LlavaQwen2ForCausalLM.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 4097, in from_pretrained model = cls(config, *model_args, **model_kwargs) File "/home/user/app/modeling_llava_qwen2.py", line 2240, in __init__ self.model = LlavaQwen2Model(config) File "/home/user/app/modeling_llava_qwen2.py", line 2232, in __init__ super(LlavaQwen2Model, self).__init__(config) File "/home/user/app/modeling_llava_qwen2.py", line 610, in __init__ super(LlavaMetaModel, self).__init__(config) File "/home/user/app/modeling_llava_qwen2.py", line 1749, in __init__ self.embed_tokens = nn.Embedding(config.vocab_size, config.hidden_size, self.padding_idx) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/sparse.py", line 167, in __init__ torch.empty((num_embeddings, embedding_dim), **factory_kwargs), File "/usr/local/lib/python3.10/site-packages/spaces/zero/torch/patching.py", line 184, in __torch_function__ res = func(*args_, **kwargs_) File "/usr/local/lib/python3.10/site-packages/torch/utils/_device.py", line 106, in __torch_function__ return func(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 319, in _lazy_init torch._C._cuda_init() File "/usr/local/lib/python3.10/site-packages/spaces/zero/torch/patching.py", line 269, 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: