athrael-soju/vdr-train-500k
Viewer • Updated • 651k • 27 • 1
How to use athrael-soju/BiUltraVBERT with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("athrael-soju/BiUltraVBERT", device_map="auto")A single-vector retriever for visual documents: UltraVBERT with a mean-pooled 768-d embedding per page and query.
Coming soon!
| Model | Role | Status |
|---|---|---|
| UltraVBERT | Base encoder aligned to page images | Available |
| BiUltraVBERT | Single-vector retriever for visual documents | Cooking |
| ColUltraVBERT | Late-interaction retriever for visual documents | Available · Demo |