clear prompt cache for training

2025-03-04 15:14:22 -08:00 · 2025-03-04 15:14:22 -08:00 · 96ba683181
commit 96ba683181
parent c552efe659
1 changed files with 1 additions and 0 deletions
--- a/src/mflux/dreambooth/dreambooth.py
+++ b/src/mflux/dreambooth/dreambooth.py
@ -62,6 +62,7 @@ class DreamBooth:
                )  # fmt: off
                image.save(path=training_state.get_current_validation_image_path(training_spec))
                del image
+                flux.prompt_cache = {}

            # Save checkpoint periodically
            if training_state.should_save(training_spec):