github/autoMate
1
0
Fork 0
mirror of https://github.com/yuruotong1/autoMate.git synced 2026-03-22 13:07:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

delete flo

Browse Source
This commit is contained in:
yuruo
2025-03-19 22:06:52 +08:00
parent a48ff2d37d
commit d8d4a1fb5b
2 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
gradio_ui/app.py
View File

@@ -14,7 +14,7 @@ from gradio_ui.loop import (
import base64 import base64
from xbrain.utils.config import Config from xbrain.utils.config import Config
from util.download_weights import OMNI_PARSER_DIR, FLORENCE_DIR from util.download_weights import OMNI_PARSER_DIR
CONFIG_DIR = Path("~/.anthropic").expanduser() CONFIG_DIR = Path("~/.anthropic").expanduser()
API_KEY_FILE = CONFIG_DIR / "api_key" API_KEY_FILE = CONFIG_DIR / "api_key"

5
util/download_weights.py
View File

@@ -3,7 +3,6 @@ from pathlib import Path
__WEIGHTS_DIR = Path("weights") __WEIGHTS_DIR = Path("weights")
OMNI_PARSER_DIR = os.path.join(__WEIGHTS_DIR, "AI-ModelScope", "OmniParser-v2___0") OMNI_PARSER_DIR = os.path.join(__WEIGHTS_DIR, "AI-ModelScope", "OmniParser-v2___0")
FLORENCE_DIR = os.path.join(__WEIGHTS_DIR, "AI-ModelScope", "Florence-2-base-ft")
def download(): def download():
from modelscope import snapshot_download from modelscope import snapshot_download
# Create weights directory # Create weights directory
@@ -13,9 +12,5 @@ def download():
cache_dir='weights', cache_dir='weights',
) )
snapshot_download(
'AI-ModelScope/Florence-2-base-ft',
cache_dir='weights' )
if __name__ == "__main__": if __name__ == "__main__":
download() download()