Move VAE checkpoints into vae/

README.md

@@ -49,7 +49,7 @@ DDCONFIG = dict(double_z=True, z_channels=16, resolution=256, in_channels=3, out
 encoder = Encoder(**DDCONFIG)
 decoder = Decoder(**{**DDCONFIG, "attn_resolutions": []})   # no attention in the decoder
 
-sd = torch.load("kl16_mm3ch_gan_balanced_130k.ckpt", map_location="cpu")["model"]
+sd = torch.load("vae/kl16_mm3ch_gan_balanced_130k.ckpt", map_location="cpu")["model"]
 # keys: encoder.* / decoder.* / quant_conv.* / post_quant_conv.*
 ```