runtime error

Exit code: 1. Reason: , 55.9MB/s] data-00002-of-00005.arrow: 100%|██████████| 407M/407M [00:02<00:00, 199MB/s] data-00003-of-00005.arrow: 0%| | 0.00/407M [00:00<?, ?B/s] data-00003-of-00005.arrow: 17%|█▋ | 70.4M/407M [00:01<00:06, 53.7MB/s] data-00003-of-00005.arrow: 100%|██████████| 407M/407M [00:02<00:00, 202MB/s] data-00004-of-00005.arrow: 0%| | 0.00/404M [00:00<?, ?B/s] data-00004-of-00005.arrow: 16%|█▋ | 66.5M/404M [00:01<00:06, 52.8MB/s] data-00004-of-00005.arrow: 100%|██████████| 404M/404M [00:02<00:00, 178MB/s]  data-00004-of-00005.arrow: 100%|██████████| 404M/404M [00:02<00:00, 159MB/s] Generating train split: 0%| | 0/13312 [00:00<?, ? examples/s] Generating train split: 40%|███▉ | 5263/13312 [00:01<00:01, 5242.14 examples/s] Generating train split: 100%|██████████| 13312/13312 [00:01<00:00, 9469.12 examples/s] Traceback (most recent call last): File "/home/user/app/train_lora.py", line 49, in <module> loss = pipe.unet(images) # 简化写法，实际需做noise prediction File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1773, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1784, in _call_impl return forward_call(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/peft/peft_model.py", line 881, in forward return self.get_base_model()(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1773, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1784, in _call_impl return forward_call(*args, **kwargs) TypeError: UNet2DConditionModel.forward() missing 2 required positional arguments: 'timestep' and 'encoder_hidden_states'

Container logs: