changed model to MiaoshouAI/Florence-2-base-PromptGen-v2.0

regiellis · Nov 7, 2024 · 8193d48 · 8193d48
1 parent 77fdc05
commit 8193d48
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,10 @@
 
 ## All notable changes to this project will be documented in this file
 
+
+##  [1.5.0] - 2024-10-07
+- **Model Update**: replaced Florence with a new model, MiaoshouAI/Florence-2-base-PromptGen-v2.0. This model is fine-tune of Florence2
+
 ##  [1.4.1] - 2024-10-07
 - **Hotfix**: Update caption mode for JoyCaption, fix some typoes
 

diff --git a/ecko_cli/__version__.py b/ecko_cli/__version__.py
@@ -1 +1 @@
-__version__ = "1.4.1"
+__version__ = "1.5.0"
diff --git a/ecko_cli/analyze.py b/ecko_cli/analyze.py
@@ -26,7 +26,7 @@
     smart_resize,
 )
 
-CAPTION_MODEL = "microsoft/Florence-2-large"
+CAPTION_MODEL = "MiaoshouAI/Florence-2-base-PromptGen-v2.0" #"microsoft/Florence-2-large" #"MiaoshouAI/Florence-2-large-PromptGen-v2.0" 
 JOYCAP_MODEL = "fancyfeast/llama-joycaption-alpha-two-hf-llava"
 IMAGE_SIZE_JOYCAP = (384, 384)
 

diff --git a/ecko_cli/cli.py b/ecko_cli/cli.py
@@ -518,3 +518,13 @@ def install_flash_attention():
         "pip install flash-attn --no-build-isolation",
         shell=True,
     )
+
+@tools_cli.command()
+def install_models():
+    """
+    Install required models
+    """
+    subprocess.run(
+        "hf download MiaoshouAI/Florence-2-base-PromptGen-v2.0",
+        shell=True,
+    )