add pruned vae

vae/kl-f8-anime2.ckpt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:72da9f22a79b6588565515084ba6ad0c0a16653cfc78cce3a68b3914f32079af
-size 1096241754
+oid sha256:df3c506e51b7ee1d7b5a6a2bb7142d47d488743c96aa778afb0f53a2cdc2d38d
+size 404662241

vae/pruner.py

@@ -0,0 +1,15 @@
+import torch
+import argparse
+
+parser = argparse.ArgumentParser()
+parser.add_argument('--input', '-I', type=str, help='Input file to prune', required = True)
+args = parser.parse_args()
+file = args.input
+
+checkpoint = torch.load(file)
+new_sd = dict()
+for k in checkpoint.keys():
+    if k != 'optimizer_states':
+        new_sd[k] = checkpoint[k]
+
+torch.save(new_sd, f'pruned-{file}')