ubuntu上双卡多卡Dreambooth训练报错

steps: 0%| | 0/234 [00:00
[rank1]: train(args)
[rank1]: File "/workspace/lora-scripts/./scripts/stable/train_db.py", line 343, in train
[rank1]: encoder_hidden_states = train_util.get_hidden_states(
[rank1]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank1]: File "/workspace/lora-scripts/scripts/stable/library/train_util.py", line 4427, in get_hidden_states
[rank1]: encoder_hidden_states = text_encoder.text_model.final_layer_norm(encoder_hidden_states)
[rank1]: ^^^^^^^^^^^^^^^^^^^^^^^
[rank1]: File "/root/miniconda3/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1729, in __getattr__
[rank1]: raise AttributeError(f"'{type(self).__name__}' object has no attribute '{name}'")
[rank1]: AttributeError: 'DistributedDataParallel' object has no attribute 'text_model'
[rank0]: Traceback (most recent call last):
[rank0]: File "/workspace/lora-scripts/./scripts/stable/train_db.py", line 529, in
[rank0]: train(args)
[rank0]: File "/workspace/lora-scripts/./scripts/stable/train_db.py", line 343, in train
[rank0]: encoder_hidden_states = train_util.get_hidden_states(
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/workspace/lora-scripts/scripts/stable/library/train_util.py", line 4427, in get_hidden_states
[rank0]: encoder_hidden_states = text_encoder.text_model.final_layer_norm(encoder_hidden_states)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/root/miniconda3/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1729, in __getattr__
[rank0]: raise AttributeError(f"'{type(self).__name__}' object has no attribute '{name}'")
[rank0]: AttributeError: 'DistributedDataParallel' object has no attribute 'text_model'
steps: 0%| | 0/234 [00:01", line 198, in _run_module_as_main
File "", line 88, in _run_code
File "/root/miniconda3/lib/python3.12/site-packages/accelerate/commands/launch.py", line 1116, in
main()
File "/root/miniconda3/lib/python3.12/site-packages/accelerate/commands/launch.py", line 1112, in main
launch_command(args)
File "/root/miniconda3/lib/python3.12/site-packages/accelerate/commands/launch.py", line 1097, in launch_command
multi_gpu_launcher(args)
File "/root/miniconda3/lib/python3.12/site-packages/accelerate/commands/launch.py", line 734, in multi_gpu_launcher
distrib_run.run(args)
File "/root/miniconda3/lib/python3.12/site-packages/torch/distributed/run.py", line 892, in run
elastic_launch(
File "/root/miniconda3/lib/python3.12/site-packages/torch/distributed/launcher/api.py", line 133, in __call__
return launch_agent(self._config, self._entrypoint, list(args))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/root/miniconda3/lib/python3.12/site-packages/torch/distributed/launcher/api.py", line 264, in launch_agent
raise ChildFailedError(
torch.distributed.elastic.multiprocessing.errors.ChildFailedError:
============================================================
./scripts/stable/train_db.py FAILED
------------------------------------------------------------
Failures:

------------------------------------------------------------
Root Cause (first observed failure):
[0]:
time : 2024-10-10_02:41:57
host : ubuntu-Super-Server
rank : 1 (local_rank: 1)
exitcode : 1 (pid: 3148)
error_file:
traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html
============================================================
02:41:58-146594 ERROR Training failed / 训练失败