Fix

2 files changed, 0 insertions, 2 deletions

diff --git a/dreambooth.py b/dreambooth.py
index a266d83..d15f1ee 100644
--- a/dreambooth.py
+++ b/dreambooth.py
@@ -538,7 +538,6 @@ class Checkpointer:
 
         del unwrapped_unet
         del unwrapped_text_encoder
-        del scheduler
         del pipeline
         del generator
         del stable_latents
diff --git a/textual_inversion.py b/textual_inversion.py
index 40ddaab..b676088 100644
--- a/textual_inversion.py
+++ b/textual_inversion.py
@@ -474,7 +474,6 @@ class Checkpointer:
                 del image_grid
 
         del unwrapped
-        del scheduler
         del pipeline
         del generator
         del stable_latents