stage3测试问题

[1llss]

作者您好。我在运行stage3_batchtest_refined_model.py的时候遇到了下面的错误。这似乎是一个模型架构的问题，请问您能否为我解答一下。
vae_gen_t_image shape: torch.Size([1, 3, 512, 512])
Original gen_t_img_f shape: torch.Size([1, 4, 64, 64])
0%| | 0/20 [00:00<?, ?it/s]vae_gen_t_image shape: torch.Size([1, 3, 512, 512])
0%| | 0/20 [00:00<?, ?it/s]
Process Process-1:
Traceback (most recent call last):
File "D:\ana\envs\PCDMs\lib\multiprocessing\process.py", line 314, in _bootstrap
self.run()
File "D:\ana\envs\PCDMs\lib\multiprocessing\process.py", line 108, in run
self._target(*self._args, **self._kwargs)
File "D:\edge下载文件\姿态识别\PCDMs-main\stage3_batchtest_refined_model.py", line 161, in inference
output = pipe(
File "D:\ana\envs\PCDMs\lib\site-packages\torch\autograd\grad_mode.py", line 27, in decorate_context
return func(*args, **kwargs)
File "D:\edge下载文件\姿态识别\PCDMs-main\src\pipelines\stage3_refined_pipeline.py", line 547, in call
noise_mask_maskedimage_latents = torch.cat([latent_model_input, gen_t_img_f], dim=1).to(dtype=torch.float32)
RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 8 but got size 2 for tensor number 1 in the list.
Original gen_t_img_f shape: torch.Size([1, 4, 64, 64])
0%| | 0/20 [00:00<?, ?it/s]
Process Process-2:
Traceback (most recent call last):
File "D:\ana\envs\PCDMs\lib\multiprocessing\process.py", line 314, in _bootstrap
self.run()
File "D:\ana\envs\PCDMs\lib\multiprocessing\process.py", line 108, in run
self._target(*self._args, **self._kwargs)
File "D:\edge下载文件\姿态识别\PCDMs-main\stage3_batchtest_refined_model.py", line 161, in inference
output = pipe(
File "D:\ana\envs\PCDMs\lib\site-packages\torch\autograd\grad_mode.py", line 27, in decorate_context
return func(*args, **kwargs)
File "D:\edge下载文件\姿态识别\PCDMs-main\src\pipelines\stage3_refined_pipeline.py", line 547, in call
noise_mask_maskedimage_latents = torch.cat([latent_model_input, gen_t_img_f], dim=1).to(dtype=torch.float32)
RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 8 but got size 2 for tensor number 1 in the list.

[muzishen]

What is the batchsize and the sample number