Update celebhq.yaml

explainingai-code · web-flow · commit 250cf513c331 · 2024-09-23T22:11:24.000+05:30
diff --git a/config/celebhq.yaml b/config/celebhq.yaml
@@ -23,7 +23,7 @@ ldm_params:
   num_up_layers : 2
 
 autoencoder_params:
-  z_channels: 3
+  z_channels: 4
   codebook_size : 8192
   down_channels : [64, 128, 256, 256]
   mid_channels : [256, 256]
