Merge pull request #136 from DanaCase/clear_prompt_cache

clear prompt cache for training
2025-03-05 07:30:16 +01:00 · 2025-03-05 07:30:16 +01:00 · 552eb6a3bb
commit 552eb6a3bb
parent ceea386361 96ba683181
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):