tiny changes

2024-09-12 10:06:38 +02:00
parent c1916918a1
commit 4df66e932d
1 changed files with 2 additions and 1 deletions
@@ -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(