The size of tensor a (368) must match the size of tensor b (369) at non-singleton dimension 3
Traceback (most recent call last):                                                             | 0/330 [00:00<?, ?it/s]File “D:\Charis\img2img-turbo-main\src\train_cyclegan_turbo.py”, line 389, inmain(args)File “D:\Charis\img2img-turbo-main\src\train_cyclegan_turbo.py”, line 329, in maineval_fake_b = CycleGAN_Turbo.forward_with_networks(img_a, “a2b”, eval_vae_enc, eval_unet,File “D:\Charis\img2img-turbo-main\src\cyclegan_turbo.py”, line 206, in forward_with_networksx_out_decoded = vae_dec(x_out, direction=direction)File “C:\Users\LocalAdmin\anaconda3\envs\img2img\lib\site-packages\torch\nn\modules\module.py”, line 1501, in _call_implreturn forward_call(*args, **kwargs)File “C:\Users\LocalAdmin\anaconda3\envs\img2img\lib\site-packages\accelerate\utils\operations.py”, line 819, in forwardreturn model_forward(*args, **kwargs)File “C:\Users\LocalAdmin\anaconda3\envs\img2img\lib\site-packages\accelerate\utils\operations.py”, line 807, incallreturn convert_to_fp32(self.model_forward(*args, **kwargs))File “C:\Users\LocalAdmin\anaconda3\envs\img2img\lib\site-packages\torch\amp\autocast_mode.py”, line 14, in decorate_autocastreturn func(*args, **kwargs)File “D:\Charis\img2img-turbo-main\src\cyclegan_turbo.py”, line 44, in forwardx_decoded = (_vae.decode(x / _vae.config.scaling_factor).sample).clamp(-1, 1)File “C:\Users\LocalAdmin\anaconda3\envs\img2img\lib\site-packages\diffusers\utils\accelerate_utils.py”, line 46, in wrapperreturn method(self, *args, **kwargs)File “C:\Users\LocalAdmin\anaconda3\envs\img2img\lib\site-packages\diffusers\models\autoencoders\autoencoder_kl.py”, line 304, in decodedecoded = self._decode(z).sampleFile “C:\Users\LocalAdmin\anaconda3\envs\img2img\lib\site-packages\diffusers\models\autoencoders\autoencoder_kl.py”, line 275, in _decodedec = self.decoder(z)File “C:\Users\LocalAdmin\anaconda3\envs\img2img\lib\site-packages\torch\nn\modules\module.py”, line 1501, in _call_implreturn forward_call(*args, **kwargs)File “D:\Charis\img2img-turbo-main\src\model.py”, line 44, in my_vae_decoder_fwdsample = sample + skip_inRuntimeError: The size of tensor a (368) must match the size of tensor b (369) at non-singleton dimension 3