diff --git a/src/mflux/dreambooth/dreambooth.py b/src/mflux/dreambooth/dreambooth.py
index cd45f3b..16d8cf8 100644
--- 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):