from transformers import AutoModelForCausalLM, AutoProcessor
from peft import PeftModel

base = AutoModelForCausalLM.from_pretrained("allenai/olmOCR-2-7B-1025")
model = PeftModel.from_pretrained(base, "rehan953/olmOCR-2-7B-1025-finetuned_v2")
processor = AutoProcessor.from_pretrained("allenai/olmOCR-2-7B-1025")
# Then run inference as usual with processor + model

Base model

Base: allenai/olmOCR-2-7B-1025

Training

Trained with Allen AI olmocr (LoRA). Use this adapter for inference only; no optimizer state is included.