Update codeformer_arch.py

codeformer_arch.py

@@ -140,7 +140,7 @@ class CodeFormer(VQAutoEncoder):
                  connect_list=['32', '64', '128', '256'],
                  fix_modules=['quantize', 'decoder'], vqgan_path=None):
         down_factor = [1, 2, 2, 4, 4, 8]
-        super().__init__(512, 64, down_factor, 'nearest', len(down_factor) - 1, 256, codebook_size)  # Changed z_channels from 16 to 256
+        super().__init__(3, 64, down_factor, 'nearest', len(down_factor) - 1, 256, codebook_size)  # Changed in_channel from 512 to 3
 
         if vqgan_path is not None:
             self.load_state_dict(torch.load(vqgan_path, map_location='cpu')['params_ema'])