yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-11-25 06:41:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes to train.py

Browse Source
This commit is contained in:
Jake Poznanski 2025-10-28 22:15:45 +00:00
parent 91962d64e2
commit 4768db63d2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
olmocr/train/train.py
View File

@ -271,7 +271,7 @@ def main():
# Load model
logger.info(f"Loading model: {config.model.name}")
if "qwen2.5-vl" in config.model.name.lower():
if "qwen2.5-vl" in config.model.name.lower() or "olmocr-2-7b-1025" in config.model.name.lower():
model_class = Qwen2_5_VLForConditionalGeneration
model = model_class.from_pretrained(config.model.name, **model_init_kwargs)
elif "qwen2-vl" in config.model.name.lower():