runtime error

Exit code: 1. Reason: ng.py", line 198, in __torch_function__ return transform_subclass(wrapper_tensor, lambda _, t: func(t)) File "/usr/local/lib/python3.10/site-packages/torch/utils/_python_dispatch.py", line 260, in transform_subclass assert sub.shape == outer_size, \ AssertionError: Expected return value from <class 'torch.nested._internal.nested_tensor.NestedTensor'>__tensor_unflatten__() to have shape equal to (1, j0, 3), but got: (1, j1, 3) The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/home/user/app/app.py", line 1, in <module> from model_worker import ModelWorker File "/home/user/app/model_worker.py", line 19, in <module> from mplug_docowl.utils import (build_logger, server_error_msg, File "/home/user/app/mplug_docowl/__init__.py", line 1, in <module> from .model import MPLUGDocOwlLlamaForCausalLM File "/home/user/app/mplug_docowl/model/__init__.py", line 1, in <module> from .modeling_mplug_docowl import MPLUGDocOwlLlamaForCausalLM File "/home/user/app/mplug_docowl/model/modeling_mplug_docowl.py", line 22, in <module> from transformers import AutoConfig, AutoModelForCausalLM, LlamaConfig, LlamaModel, LlamaForCausalLM File "<frozen importlib._bootstrap>", line 1075, in _handle_fromlist File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1090, in __getattr__ value = getattr(module, name) File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1089, in __getattr__ module = self._get_module(self._class_to_module[name]) File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1101, in _get_module raise RuntimeError( RuntimeError: Failed to import transformers.models.llama.modeling_llama because of the following error (look up to see its traceback): Expected return value from <class 'torch.nested._internal.nested_tensor.NestedTensor'>__tensor_unflatten__() to have shape equal to (1, j0, 3), but got: (1, j1, 3)

Container logs: