Upload 3 files

vae/kl-f8-anime.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2f11c4a99ddc28d0ad8bce0acc38bed310b45d38a3fe4bb367dc30f3ef1a4868
+size 404657953

vae/kl-f8-anime2.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+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}')