Upload folder using huggingface_hub

README.md

@@ -117,8 +117,7 @@ detailed, but uses more VRAM.
 
 | Model | Size | Best for |
 | --- | --- | --- |
-| **florence2-large-ft** (default) | ~1.5 GB | Best quality. Needs a normal GPU. |
-| florence2-large | ~1.5 GB | Slightly less accurate. |
+| **florence2-large** (default) | ~1.5 GB | Slightly less accurate. Needs a normal GPU. |
 | **florence2-base** | ~0.45 GB | **2 GB GPUs**, or when you want it fast. |
 
 If your card only has 2 GB of VRAM, choose **florence2-base**.

caption.py

@@ -42,13 +42,12 @@ CONFIG_PATH = Path(__file__).parent / "caption_config.json"
 
 # Model choices. Sizes are approximate fp16 GPU footprint.
 MODELS = {
-    "florence2-large-ft": ("microsoft/Florence-2-large-ft", "~1.5 GB — best quality (recommended)"),
     "florence2-large":    ("microsoft/Florence-2-large",    "~1.5 GB — base, slightly less accurate"),
     "florence2-base":     ("microsoft/Florence-2-base",     "~0.45 GB — fast, great for 2 GB GPUs"),
 }
 
 DEFAULT_CONFIG = {
-    "model": "microsoft/Florence-2-large-ft",
+    "model": "microsoft/Florence-2-large",
     "prefix": "",
     "suffix": "",
     "max_new_tokens": 256,

caption_config.json

@@ -1,5 +1,5 @@
 {
-  "model": "microsoft/Florence-2-large-ft",
+  "model": "microsoft/Florence-2-large",
   "prefix": "",
   "suffix": "",
   "max_new_tokens": 256,