From f49a07f1059e88ddfaa3f3f2d63dbfc7a44983e3 Mon Sep 17 00:00:00 2001
From: Boris Dayma <boris.dayma@gmail.com>
Date: Thu, 24 Mar 2022 15:11:47 +0000
Subject: [PATCH] feat: remove print statement

---
 vqgan_jax/modeling_flax_vqgan.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/vqgan_jax/modeling_flax_vqgan.py b/vqgan_jax/modeling_flax_vqgan.py
index 3792960..4e3d681 100644
--- a/vqgan_jax/modeling_flax_vqgan.py
+++ b/vqgan_jax/modeling_flax_vqgan.py
@@ -399,8 +399,6 @@ def setup(self):
                                                     - 1]
     curr_res = self.config.resolution // 2**(self.config.num_resolutions - 1)
     self.z_shape = (1, self.config.z_channels, curr_res, curr_res)
-    print("Working with z of shape {} = {} dimensions.".format(
-        self.z_shape, np.prod(self.z_shape)))
 
     # z to block_in
     self.conv_in = nn.Conv(