<venv>/lib/python3.10/site-packages/torch/distributed/_functional_collectives.py:22: UserWarning: Unable to import torchdynamo util `is_torchdynamo_compiling`, so won't support torchdynamo correctly 

  warnings.warn( 

Traceback (most recent call last): 

  File "<project_path>/train.py", line 27, in <module> 

    import torch_npu 

  File "<venv>/lib/python3.10/site-packages/torch_npu/__init__.py", line 14, in <module> 

    from torch.distributed.fsdp import sharded_grad_scaler 

  File "<venv>/lib/python3.10/site-packages/torch/distributed/fsdp/__init__.py", line 1, in <module> 

    from .flat_param import FlatParameter 

  File "<venv>/lib/python3.10/site-packages/torch/distributed/fsdp/flat_param.py", line 30, in <module> 

    from torch.distributed.fsdp._common_utils import ( 

  File "<venv>/lib/python3.10/site-packages/torch/distributed/fsdp/_common_utils.py", line 31, in <module> 

    from torch.distributed._tensor.device_mesh import DeviceMesh 

  File "<venv>/lib/python3.10/site-packages/torch/distributed/_tensor/__init__.py", line 319, in <module> 

    import torch.distributed._tensor._dynamo_utils 

  File "<venv>/lib/python3.10/site-packages/torch/distributed/_tensor/_dynamo_utils.py", line 1, in <module> 

    from torch._dynamo import allow_in_graph 

  File "<venv>/lib/python3.10/site-packages/torch/_dynamo/__init__.py", line 2, in <module> 

    from . import allowed_functions, convert_frame, eval_frame, resume_execution 

  File "<venv>/lib/python3.10/site-packages/torch/_dynamo/allowed_functions.py", line 24, in <module> 

    from . import config 

  File "<venv>/lib/python3.10/site-packages/torch/_dynamo/config.py", line 49, in <module> 

    torch.onnx.is_in_onnx_export: False, 

  File "<venv>/lib/python3.10/site-packages/torch/__init__.py", line 1831, in __getattr__ 

    return importlib.import_module(f".{name}", __name__) 

  File "<venv>/lib/python3.10/importlib/__init__.py", line 126, in import_module 

    return _bootstrap._gcd_import(name[level:], package, level) 

  File "<venv>/lib/python3.10/site-packages/torch/onnx/__init__.py", line 46, in <module> 

    from ._internal.exporter import (  # usort:skip. needs to be last to avoid circular import 

  File "<venv>/lib/python3.10/site-packages/torch/onnx/_internal/exporter.py", line 42, in <module> 

    from torch.onnx._internal.fx import ( 

  File "<venv>/lib/python3.10/site-packages/torch/onnx/_internal/fx/__init__.py", line 1, in <module> 

    from .patcher import ONNXTorchPatcher 

  File "<venv>/lib/python3.10/site-packages/torch/onnx/_internal/fx/patcher.py", line 11, in <module> 

    import transformers  # type: ignore[import] 

  File "<venv>/lib/python3.10/site-packages/transformers/__init__.py", line 27, in <module> 

    from . import dependency_versions_check 

  File "<venv>/lib/python3.10/site-packages/transformers/dependency_versions_check.py", line 16, in <module> 

    from .utils.versions import require_version, require_version_core 

  File "<venv>/lib/python3.10/site-packages/transformers/utils/__init__.py", line 24, in <module> 

    from .auto_docstring import ( 

  File "<venv>/lib/python3.10/site-packages/transformers/utils/auto_docstring.py", line 30, in <module> 

    from .generic import ModelOutput 

  File "<venv>/lib/python3.10/site-packages/transformers/utils/generic.py", line 465, in <module> 

    _torch_pytree.register_pytree_node( 

AttributeError: module 'torch.utils._pytree' has no attribute 'register_pytree_node'. Did you mean: '_register_pytree_node'?