LightOnOCR-2-1B-base
tabanının
12 Türkçe kurumsal belge türü × 3 çekim koşulu ile fine-tune'u.
Sayfa görüntüsünü okuma sırasına uygun, tablolar dahil
markdown olarak çıkarır.
v1 sürümü
revision="v1" altında korunur.
Genel vekaletname · DASK poliçesi · e-Arşiv fatura · Konut kira sözleşmesi ·
Banka dekontu · Tapu senedi · Maaş bordrosu · Kasko poliçesi · Araç tescil
bilgi formu · Resmî kurum yazısı · Ticaret sicil ilanı · SGK hizmet dökümü
1import torch
2from transformers import LightOnOcrForConditionalGeneration, LightOnOcrProcessor
3
4model = LightOnOcrForConditionalGeneration.from_pretrained(
5 "Werea-co/Werea-DocOCR-1B", torch_dtype=torch.bfloat16).to("cuda")
6processor = LightOnOcrProcessor.from_pretrained("Werea-co/Werea-DocOCR-1B")
7
8conv = [{"role": "user", "content": [{"type": "image", "path": "belge.jpg"}]}]
9inputs = processor.apply_chat_template(conv, add_generation_prompt=True,
10 tokenize=True, return_dict=True, return_tensors="pt")
11inputs = {k: v.to("cuda", dtype=torch.bfloat16) if v.is_floating_point() else v.to("cuda")
12 for k, v in inputs.items()}
13out = model.generate(**inputs, max_new_tokens=2048)
14print(processor.decode(out[0, inputs["input_ids"].shape[1]:], skip_special_tokens=True))