AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

delete vae and text encoders after use in SDXL training script (#6693)

Browse Source 
delete vae and text encoders after use

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
This commit is contained in:
Anatoly Belikov
2024-03-18 17:33:53 +03:00
committed by GitHub
parent 4330a747d4
commit 98d46a3f08
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/text_to_image/train_text_to_image_sdxl.py
View File

@@ -911,6 +911,7 @@ def main(args):
)
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()