Change Prunin' to Pruning

JoePenna · Nov 6, 2022 · 0339276 · 0339276
1 parent c5c48dd
commit 0339276
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/ldm/pruner.py b/ldm/pruner.py
@@ -1,5 +1,5 @@
 def prune_checkpoint(old_state):
-    print(f"Prunin' Checkpoint")
+    print(f"Pruning Checkpoint")
     pruned_checkpoint = dict()
     print(f"Checkpoint Keys: {old_state.keys()}")
     for key in old_state.keys():

diff --git a/prune_ckpt.py b/prune_ckpt.py
@@ -10,7 +10,7 @@
 ckpt = args.ckpt
 
 def prune_it(checkpoint_path):
-    print(f"Prunin' checkpoint from path: {checkpoint_path}")
+    print(f"Pruning checkpoint from path: {checkpoint_path}")
     size_initial = os.path.getsize(checkpoint_path)
     checkpoint = torch.load(checkpoint_path, map_location="cpu")
     pruned = prune_checkpoint(checkpoint)

diff --git a/scripts/prune-ckpt.py b/scripts/prune-ckpt.py
@@ -10,7 +10,7 @@
 ckpt = args.ckpt
 
 def prune_it(p, keep_only_ema=False):
-    print(f"prunin' in path: {p}")
+    print(f"Pruning in path: {p}")
     size_initial = os.path.getsize(p)
     nsd = dict()
     sd = torch.load(p, map_location="cpu")