Update iris/src/models/nets.py

iris/src/models/nets.py

@@ -102,7 +102,7 @@ class Decoder(nn.Module):
         in_ch_mult = (1,) + tuple(config["ch_mult"])
         block_in = config["ch"] * config["ch_mult"][self.num_resolutions - 1]
         curr_res = config["resolution"] // 2 ** (self.num_resolutions - 1)
-        print(f"Tokenizer : shape of latent is {config["z_channels"], curr_res, curr_res}.")
+        print(f"Tokenizer : shape of latent is {config['z_channels'], curr_res, curr_res}.")
 
         # z to block_in
         self.conv_in = torch.nn.Conv2d(config["z_channels"],