diff --git a/trainer/utils/inference.py b/trainer/utils/inference.py
index de6bb49..d6011c6 100644
--- a/trainer/utils/inference.py
+++ b/trainer/utils/inference.py
@@ -222,4 +222,4 @@ def render_images_eval(
         )
         filenames.append(filename)
 
-    return filenames
+    return filenames, validation_prompts