runtime error

Exit code: 1. Reason: ormers_modules/ucaslcl/GOT-OCR2_0/979938bf89ccdc949c0131ddd3841e24578a4742/modeling_GOT.py", line 1, in <module> from transformers import Qwen2Config, Qwen2Model, Qwen2ForCausalLM, StoppingCriteria, TextStreamer File "<frozen importlib._bootstrap>", line 1075, in _handle_fromlist File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1355, in __getattr__ value = getattr(module, name) File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1354, 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 1366, in _get_module raise RuntimeError( RuntimeError: Failed to import transformers.models.qwen2.modeling_qwen2 because of the following error (look up to see its traceback): Failed to import transformers.generation.utils because of the following error (look up to see its traceback): No module named 'torch.distributed.device_mesh' During handling of the above exception, another exception occurred: Traceback (most recent call last): File "/home/user/app/app.py", line 356, in <module> model, tokenizer, cache_manager = initialize_model_safely() File "/home/user/app/app.py", line 353, in initialize_model_safely raise Exception(f"Failed to initialize model: {str(e)}. Fallback also failed: {str(fallback_error)}") Exception: Failed to initialize model: Failed to import transformers.models.qwen2.modeling_qwen2 because of the following error (look up to see its traceback): Failed to import transformers.generation.utils because of the following error (look up to see its traceback): No module named 'torch.distributed.device_mesh'. Fallback also failed: Failed to import transformers.models.qwen2.modeling_qwen2 because of the following error (look up to see its traceback): Failed to import transformers.generation.utils because of the following error (look up to see its traceback): No module named 'torch.distributed.device_mesh'

Container logs: