Mirrors/dots.ocr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

align

Browse Source
This commit is contained in:
zhangwei13
2026-03-24 22:43:01 +08:00
parent d9ea2a4108
commit 36d7248878
59 changed files with 396 additions and 390 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/download_model.py
+3 -3
View File
@@ -5,11 +5,11 @@ import os
if __name__ == '__main__':
parser = ArgumentParser()
parser.add_argument('--type', '-t', type=str, default="huggingface")
parser.add_argument('--name', '-n', type=str, default="rednote-hilab/dots.ocr-1.5")
parser.add_argument('--name', '-n', type=str, default="rednote-hilab/dots.mocr")
args = parser.parse_args()
script_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
print(f"Attention: The model save dir dots.ocr should be replace by a name without `.` like DotsOCR, util we merge our code to transformers.")
model_dir = os.path.join(script_dir, "weights/DotsOCR_1_5")
print(f"Attention: The model save dir dots.mocr should be replace by a name without `.` like DotsMOCR, util we merge our code to transformers.")
model_dir = os.path.join(script_dir, "weights/DotsMOCR")
if not os.path.exists(model_dir):
os.makedirs(model_dir)
if args.type == "huggingface":