florence default prompt to <MORE_DETAILED_CAPTION>

matatonic · Jun 19, 2024 · 2beb5e0 · 2beb5e0
1 parent d087b6c
commit 2beb5e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vision_qna.py b/vision_qna.py
@@ -665,7 +665,7 @@ async def glm4v_prompt_from_messages(messages: list[Message], img_tok = "<|begin
     return images, prompt
 
 async def florence_prompt_from_messages(messages: list[Message], url_handler = url_to_image):
-    prompt = '<CAPTION>' # "<CAPTION>", "<DETAILED_CAPTION>", "<MORE_DETAILED_CAPTION>", "<OCR>"
+    prompt = '<MORE_DETAILED_CAPTION>' # "<CAPTION>", "<DETAILED_CAPTION>", "<MORE_DETAILED_CAPTION>", "<OCR>"
     images = []
 
     for m in messages: