Traceback (most recent call last): | 0/330 [00:00<?, ?it/s]
File “D:\Charis\img2img-turbo-main\src\train_cyclegan_turbo.py”, line 389, in
main(args)
File “D:\Charis\img2img-turbo-main\src\train_cyclegan_turbo.py”, line 329, in main
eval_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_networks
x_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_impl
return forward_call(*args, **kwargs)
File “C:\Users\LocalAdmin\anaconda3\envs\img2img\lib\site-packages\accelerate\utils\operations.py”, line 819, in forward
return model_forward(*args, **kwargs)
File “C:\Users\LocalAdmin\anaconda3\envs\img2img\lib\site-packages\accelerate\utils\operations.py”, line 807, in call
return 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_autocast
return func(*args, **kwargs)
File “D:\Charis\img2img-turbo-main\src\cyclegan_turbo.py”, line 44, in forward
x_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 wrapper
return method(self, *args, **kwargs)
File “C:\Users\LocalAdmin\anaconda3\envs\img2img\lib\site-packages\diffusers\models\autoencoders\autoencoder_kl.py”, line 304, in decode
decoded = self._decode(z).sample
File “C:\Users\LocalAdmin\anaconda3\envs\img2img\lib\site-packages\diffusers\models\autoencoders\autoencoder_kl.py”, line 275, in _decode
dec = self.decoder(z)
File “C:\Users\LocalAdmin\anaconda3\envs\img2img\lib\site-packages\torch\nn\modules\module.py”, line 1501, in _call_impl
return forward_call(*args, **kwargs)
File “D:\Charis\img2img-turbo-main\src\model.py”, line 44, in my_vae_decoder_fwd
sample = sample + skip_in
RuntimeError: The size of tensor a (368) must match the size of tensor b (369) at non-singleton dimension 3