From 4df66e932dcc36cd6647772bca03cfb1f5227ff2 Mon Sep 17 00:00:00 2001
From: xander <xandersteenbruggex@gmail.com>
Date: Thu, 12 Sep 2024 10:06:38 +0200
Subject: [PATCH] tiny changes

---
 trainer/preprocess.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/trainer/preprocess.py b/trainer/preprocess.py
index 940b50f..213a8f2 100755
--- a/trainer/preprocess.py
+++ b/trainer/preprocess.py
@@ -525,7 +525,8 @@ def florence_caption_dataset(images, captions):
         if captions[i] is None:
             #prompt = random.choice(["<CAPTION>", "<DETAILED_CAPTION>"])
             prompt = "<CAPTION>"
-            #prompt = "<DETAILED_CAPTION>"
+            prompt = "<DETAILED_CAPTION>"
+            prompt = "<MORE_DETAILED_CAPTION>"
 
             inputs = processor(text=prompt, images=image, return_tensors="pt").to(device, torch_dtype)
             generated_ids = model.generate(