From 96ba68318116e2f5adab926995624a6d04d1aa7a Mon Sep 17 00:00:00 2001
From: Dana Case <case.danal@gmail.com>
Date: Tue, 4 Mar 2025 15:14:22 -0800
Subject: [PATCH] clear prompt cache for training

---
 src/mflux/dreambooth/dreambooth.py | 1 +
 1 file changed, 1 insertion(+)

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):