jiuface commited on
Commit
6aa2176
1 Parent(s): f62183e
utils/__pycache__/florence.cpython-310.pyc CHANGED
Binary files a/utils/__pycache__/florence.cpython-310.pyc and b/utils/__pycache__/florence.cpython-310.pyc differ
 
utils/florence.py CHANGED
@@ -8,7 +8,7 @@ from transformers import AutoModelForCausalLM, AutoProcessor
8
  from transformers.dynamic_module_utils import get_imports
9
 
10
  # FLORENCE_CHECKPOINT = "microsoft/Florence-2-base"
11
- FLORENCE_CHECKPOINT = "microsoft/Florence-2-large"
12
  FLORENCE_OBJECT_DETECTION_TASK = '<OD>'
13
  FLORENCE_DETAILED_CAPTION_TASK = '<MORE_DETAILED_CAPTION>'
14
  FLORENCE_CAPTION_TO_PHRASE_GROUNDING_TASK = '<CAPTION_TO_PHRASE_GROUNDING>'
 
8
  from transformers.dynamic_module_utils import get_imports
9
 
10
  # FLORENCE_CHECKPOINT = "microsoft/Florence-2-base"
11
+ FLORENCE_CHECKPOINT = "microsoft/Florence-2-large-ft"
12
  FLORENCE_OBJECT_DETECTION_TASK = '<OD>'
13
  FLORENCE_DETAILED_CAPTION_TASK = '<MORE_DETAILED_CAPTION>'
14
  FLORENCE_CAPTION_TO_PHRASE_GROUNDING_TASK = '<CAPTION_TO_PHRASE_GROUNDING>'