VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

move releasing cache outside of the loop

Browse Source
This commit is contained in:
Kohya S
2023-07-30 13:30:42 +09:00
parent 54a4aa22ed
commit e6034b7eb6
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
library/train_util.py
View File

@@ -2165,8 +2165,9 @@ def cache_batch_latents(
info.latents = latent info.latents = latent
if flip_aug: if flip_aug:
info.latents_flipped = flipped_latent info.latents_flipped = flipped_latent
if torch.cuda.is_available():
torch.cuda.empty_cache() if torch.cuda.is_available():
torch.cuda.empty_cache()
def cache_batch_text_encoder_outputs( def cache_batch_text_encoder_outputs(