Views
No views yet
| Dataset | m4xi/unimer-merged (~1.04M train samples, 98/2 train/val split) |
| LoRA rank / alpha | 16 / 32 |
| Target modules | q/k/v/o/gate/up/down_proj (vision encoder frozen) |
| Effective batch size | 16 (4 per device × 4 grad accum) |
| Learning rate | 2e-4, cosine decay, 3% warmup |
| Precision | bf16 |
| Steps | 115,500 (~1.78 epochs, ~1.85M samples seen) |
| Hardware | NVIDIA L40S |
1from transformers import AutoModel, AutoTokenizer
2from peft import PeftModel
3
4model = AutoModel.from_pretrained("stepfun-ai/GOT-OCR2_0", trust_remote_code=True, dtype=torch.bfloat16)
5model = PeftModel.from_pretrained(model, "maximuskiii/got-mer-lora-r16")
6model = model.merge_and_unload()