github/autoMate
1
0
Fork 0
mirror of https://github.com/yuruotong1/autoMate.git synced 2026-03-22 04:57:18 +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

5
util/download_weights.py
View File

@@ -3,7 +3,6 @@ from pathlib import Path
__WEIGHTS_DIR = Path("weights")
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():
from modelscope import snapshot_download
# Create weights directory
@@ -13,9 +12,5 @@ def download():
cache_dir='weights',
)
snapshot_download(
'AI-ModelScope/Florence-2-base-ft',
cache_dir='weights' )
if __name__ == "__main__":
download()