From bf91ee3e83e1f677fe521ba92f94cf8a199ad7bc Mon Sep 17 00:00:00 2001
From: mayukhdeb <mayukhmainak2000@gmail.com>
Date: Thu, 4 Apr 2024 02:30:37 -0700
Subject: [PATCH] return filenames and prompts

---
 trainer/utils/inference.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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