From 4768db63d2e2f14c4134e50013d25f18ccb32ff0 Mon Sep 17 00:00:00 2001 From: Jake Poznanski Date: Tue, 28 Oct 2025 22:15:45 +0000 Subject: [PATCH] Fixes to train.py --- olmocr/train/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/olmocr/train/train.py b/olmocr/train/train.py index 23f510f..35150fd 100644 --- a/olmocr/train/train.py +++ b/olmocr/train/train.py @@ -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():