delete vae and text encoders after use in SDXL training script (huggi…

…ngface#6693) delete vae and text encoders after use Co-authored-by: Sayak Paul <[email protected]>
tolgacangoz · Mar 18, 2024 · 98d46a3 · 98d46a3
1 parent 4330a74
commit 98d46a3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/text_to_image/train_text_to_image_sdxl.py b/examples/text_to_image/train_text_to_image_sdxl.py
@@ -911,6 +911,7 @@ def preprocess_train(examples):
         )
         precomputed_dataset = precomputed_dataset.with_transform(preprocess_train)
 
+    del compute_vae_encodings_fn, compute_embeddings_fn, text_encoder_one, text_encoder_two
     del text_encoders, tokenizers, vae
     gc.collect()
     torch.cuda.empty_cache()