Update config.py

config.py

@@ -11,7 +11,7 @@ from diffuse.neural_network import CondUNet2D
 
 # Model definition
 model = CondUNet2D(
-    in_channels=IN_CHANNELS,
+    in_channels=32,
     ch=32,  # base_channels
     ch_mult=(1, 2, 2),  # channel_multipliers
     num_res_blocks=1,