allenai / allenai/olmocr

Make the model run on lower GPU at the expense of time/speed

Ouverte
#458 0 commentaires 0 réactions 0 personnes assignées Voir sur GitHub
Langage dominant
Python
Étoiles
19.5k
Forks
1.6k
Métriques de merge des PR
Aucune PR mergée en 30 j

Description

### 🚀 The feature, motivation and pitch

Add command-line flags (e.g., --max-model-len) that allow users to:
Manually cap the context window (e.g., to 4096 or 8192) to fit within smaller VRAM buffers.

Trade off processing speed or context size to allow the model to run on 12GB, 10GB, or 8GB GPUs.

This would make the tool accessible to a much wider range of researchers and hobbyists who are running local workstations rather than server-grade hardware.

### Alternatives

_No response_

### Additional context

_No response_

Guide de contribution

Ouvrir le guide de contribution

Évaluation

Cette issue n'a pas encore été évaluée.

Recevez les nouvelles issues par e-mail

Un résumé court des issues GitHub adaptées aux débutants.