monster/Qwen-Image-Layered-MRP-MLX
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

clear prompt cache for training

Browse Source
This commit is contained in:
Dana Case 2025-03-04 15:14:22 -08:00
parent c552efe659
commit 96ba683181
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/mflux/dreambooth/dreambooth.py
View File

@ -62,6 +62,7 @@ class DreamBooth:
) # fmt: off ) # fmt: off
image.save(path=training_state.get_current_validation_image_path(training_spec)) image.save(path=training_state.get_current_validation_image_path(training_spec))
del image del image
flux.prompt_cache = {}
# Save checkpoint periodically # Save checkpoint periodically
if training_state.should_save(training_spec): if training_state.should_save(training_spec):