Revert "Save captions with checkpoint"

JoePenna · Dec 16, 2022 · f00062a · f00062a
1 parent a1781aa
commit f00062a
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 42 deletions.
diff --git a/ldm/modules/callbacks/captions.py b/ldm/modules/callbacks/captions.py
diff --git a/ldm/pruner.py b/ldm/pruner.py
@@ -1,6 +1,7 @@
 def prune_checkpoint(old_state):
     print(f"Pruning Checkpoint")
     pruned_checkpoint = dict()
+    print(f"Checkpoint Keys: {old_state.keys()}")
     for key in old_state.keys():
         if key != "optimizer_states":
             pruned_checkpoint[key] = old_state[key]

diff --git a/main.py b/main.py
@@ -771,9 +771,6 @@ def on_train_epoch_start(self, trainer, pl_module):
             "cuda_callback": {
                 "target": "main.CUDACallback"
             },
-            "captions_callback": {
-                "target": "ldm.modules.callbacks.captions.CaptionSaverCallback"
-            }
         }
         if version.parse(pl.__version__) >= version.parse('1.4.0'):
             default_callbacks_cfg.update({'checkpoint_callback': modelckpt_cfg})

diff --git a/scripts/show_captions.py b/scripts/show_captions.py