runtime error

Exit code: 1. Reason: ", line 19, in <module> import torch.utils.checkpoint File "/usr/local/lib/python3.10/site-packages/torch/utils/checkpoint.py", line 15, in <module> from torch._functorch._aot_autograd.functional_utils import is_fun File "/usr/local/lib/python3.10/site-packages/torch/_functorch/_aot_autograd/functional_utils.py", line 19, in <module> from torch.fx.experimental.symbolic_shapes import definitely_true, sym_eq File "/usr/local/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py", line 51, in <module> from torch.fx.experimental.recording import ( File "/usr/local/lib/python3.10/site-packages/torch/fx/experimental/recording.py", line 14, in <module> trace_shape_events_log = torch._logging.getArtifactLogger( File "/usr/local/lib/python3.10/site-packages/torch/_logging/_internal.py", line 532, in getArtifactLogger ValueError: Artifact name: 'trace_shape_events' not registered,please call register_artifact('trace_shape_events') in torch._logging.registrations. The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/home/user/app/app.py", line 11, in <module> from transformers import OlmoeForCausalLM, AutoTokenizer, TextIteratorStreamer File "<frozen importlib._bootstrap>", line 1075, in _handle_fromlist File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1650, in __getattr__ value = getattr(module, name) File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1649, 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 1661, in _get_module raise RuntimeError( RuntimeError: Failed to import transformers.models.olmoe.modeling_olmoe because of the following error (look up to see its traceback): Artifact name: 'trace_shape_events' not registered,please call register_artifact('trace_shape_events') in torch._logging.registrations.

Container logs: