ValueError: 不支持的注意力模式: None

[rank0]: Traceback (most recent call last): [rank0]: File "/root/kohya_ss/sd-scripts/anima_train.py", line 769, in [rank0]: train(args) [rank0]: File "/root/kohya_ss/sd-scripts/anima_train.py", line 560, in train [rank0]: model_pred = dit( [rank0]: ^^^^ [rank0]: File "/root/kohya_ss/.venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1751, in _wrapped_call_impl [rank0]: return self._call_impl(*args, **kwargs) [rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ [rank0]: File "/root/kohya_ss/.venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl [rank0]: return forward_call(*args, **kwargs) [rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ [rank0]: File "/root/kohya_ss/.venv/lib/python3.11/site-packages/torch/nn/parallel/distributed.py", line 1637, in forward [rank0]: else self._run_ddp_forward(*inputs, **kwargs) [rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ [rank0]: File "/root/kohya_ss/.venv/lib/python3.11/site-packages/torch/nn/parallel/distributed.py", line 1464, in _run_ddp_forward [rank0]: return self.module(*inputs, **kwargs) # type: ignore[index] [rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ [rank0]: File "/root/kohya_ss/.venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1751, in _wrapped_call_impl [rank0]: return self._call_impl(*args, **kwargs) [rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ [rank0]: File "/root/kohya_ss/.venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl [rank0]: return forward_call(*args, **kwargs) [rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ [rank0]: File "/root/kohya_ss/.venv/lib/python3.11/site-packages/accelerate/utils/operations.py", line 818, in forward [rank0]: return model_forward(*args, **kwargs) [rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ [rank0]: File "/root/kohya_ss/.venv/lib/python3.11/site-packages/accelerate/utils/operations.py", line 806, in call [rank0]: return convert_to_fp32(self.model_forward(*args, **kwargs)) [rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ [rank0]: File "/root/kohya_ss/.venv/lib/python3.11/site-packages/torch/amp/autocast_mode.py", line 44, in decorate_autocast [rank0]: return func(*args, **kwargs) [rank0]: ^^^^^^^^^^^^^^^^^^^^^ [rank0]: File "/root/kohya_ss/sd-scripts/library/anima_models.py", line 1375, in forward [rank0]: return self.forward_mini_train_dit(x, timesteps, context, fps=fps, padding_mask=padding_mask, **kwargs) [rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ [rank0]: File "/root/kohya_ss/sd-scripts/library/anima_models.py", line 1353, in forward_mini_train_dit [rank0]: x_B_T_H_W_D = block(x_B_T_H_W_D, t_embedding_B_T_D, crossattn_emb, attn_params, use_fp32, **block_kwargs) [rank0]: …