runtime error

Exit code: 1. Reason: Inisialisasi model Pix2Pix... Mengunduh model dari Hugging Face Hub... Model berhasil diunduh ke: /home/user/.cache/huggingface/hub/models--Matharrr--pix2pix-coloring-manga-model/snapshots/29a12ace73ddcb0a340a031fa6f49a9a182052cb/latest_net_G.pth Membuat arsitektur model generator... initialize network with normal Memuat model dari state_dict... Traceback (most recent call last): File "/home/user/app/app.py", line 77, in <module> model = Pix2PixModel() File "/home/user/app/app.py", line 47, in __init__ self.netG.load_state_dict(state_dict) # strict=True karena input_nc sama File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 2041, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for UnetGenerator: size mismatch for model.model.3.weight: copying a param with shape torch.Size([128, 2, 4, 4]) from checkpoint, the shape in current model is torch.Size([128, 3, 4, 4]). size mismatch for model.model.3.bias: copying a param with shape torch.Size([2]) from checkpoint, the shape in current model is torch.Size([3]).

Container logs: