[VAE] fix the downsample block in Encoder. (huggingface#156)

* pass downsample_padding in encoder * update tests
yoonseokjin · Aug 6, 2022 · 4845fd1 · 4845fd1
1 parent 2c1b4e2
commit 4845fd1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/models/vae.py b/models/vae.py
@@ -40,6 +40,7 @@ def __init__(
                 out_channels=output_channel,
                 add_downsample=not is_final_block,
                 resnet_eps=1e-6,
+                downsample_padding=0,
                 resnet_act_fn=act_fn,
                 attn_num_head_channels=None,
                 temb_channels=None,