allenai / allenai/olmocr

Parallelism Problems (Only one task at a time)

Aperta
#426 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub
bug
Lingua principale
Python
Stelle
19.5k
Fork
1.6k
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

### 🐛 Describe the bug

I am running multiple nodes each running 1 L40s gpu via huggingface dedicated inference. I am passing tensor-parallel = 4, pages per group = 4, num workers = 4, max concurrent requests = 200. However, I only ever see output like this ->

Metric Name Lifetime (tokens/sec) Recently (tokens/sec)
----------------------------------------------------------------------------------
completed_pages 2.34 2.34
finished_input_tokens 3392.25 3392.25
finished_on_attempt_0 2.34 2.34
finished_output_tokens 1097.30 1097.30
server_input_tokens 3535.59 3535.59
server_output_tokens 1133.80 1133.80
2026-01-20 14:34:02,518 - __main__ - INFO -
Worker ID | finished | started
----------+----------+--------
0 | 0 | 284
1 | 24 | 284
2 | 0 | 284
3 | 0 | 284

And when i look at the load distribution, i see its all going to one node.

NOTE: I also tried running ONE node with 4 x GPUs and saw the same result.

Am I assuming that there is a mapping from worker to gpu/node...Yes. Is that wrong? And if so, are these actually valid results?

### Versions

Python 3.14.2
annotated-types==0.7.0
anthropic==0.76.0
anyio==4.12.1
attrs==25.4.0
beaker-py==2.5.3
beautifulsoup4==4.14.3
bleach==6.3.0
blinker==1.9.0
boto3==1.42.26
botocore==1.42.26
cached_path==1.8.1
certifi==2026.1.4
cffi==2.0.0
charset-normalizer==3.4.4
click==8.3.1
climage==0.2.2
cryptography==46.0.3
defusedxml==0.7.1
distro==1.9.0
docstring_parser==0.17.0
eval_type_backport==0.3.1
fastjsonschema==2.21.2
filelock==3.20.3
Flask==3.1.2
fsspec==2026.1.0
ftfy==6.3.1
fuzzysearch==0.8.1
google-api-core==2.29.0
google-auth==2.47.0
google-cloud-core==2.5.0
google-cloud-storage==3.7.0
google-crc32c==1.8.0
google-genai==1.58.0
google-resumable-media==2.8.0
googleapis-common-protos==1.72.0
greenlet==3.3.0
grpcio==1.76.0
h11==0.16.0
hf-xet==1.2.0
httpcore==1.0.9
httpx==0.28.1
huggingface-hub==0.36.0
idna==3.11
importlib_metadata==8.7.1
invoke==2.2.1
itsdangerous==2.2.0
Jinja2==3.1.6
jiter==0.12.0
jmespath==1.0.1
jsonschema==4.26.0
jsonschema-specifications==2025.9.1
jupyter_client==8.8.0
jupyter_core==5.9.1
jupyterlab_pygments==0.3.0
kdtree==0.16
lingua-language-detector==1.3.4
lxml==6.0.2
markdown-it-py==4.0.0
markdown2==2.5.4
markdownify==1.2.2
MarkupSafe==3.0.3
mdurl==0.1.2
mistralai==1.10.0
mistune==3.2.0
mpmath==1.3.0
nbclient==0.10.4
nbconvert==7.16.6
nbformat==5.10.4
networkx==3.6.1
numpy==2.4.1
olmocr==0.4.16
openai==2.15.0
opentelemetry-api==1.38.0
opentelemetry-exporter-otlp-proto-common==1.38.0
opentelemetry-exporter-otlp-proto-http==1.38.0
opentelemetry-proto==1.38.0
opentelemetry-sdk==1.38.0
opentelemetry-semantic-conventions==0.59b0
orjson==3.11.5
packaging==25.0
pandocfilters==1.5.1
pillow==10.4.0
platformdirs==4.5.1
playwright==1.57.0
proto-plus==1.27.0
protobuf==6.33.4
pyasn1==0.6.1
pyasn1_modules==0.4.2
pycparser==2.23
pydantic==2.12.5
pydantic_core==2.41.5
pyee==13.0.0
Pygments==2.19.2
pypdf==6.6.0
pypdfium2==5.3.0
python-dateutil==2.9.0.post0
python-magic==0.4.27
PyYAML==6.0.3
pyzmq==27.1.0
RapidFuzz==3.14.3
referencing==0.37.0
regex==2023.12.25
requests==2.32.5
rich==13.9.4
rpds-py==0.30.0
rsa==4.9.1
s3transfer==0.16.0
safetensors==0.7.0
sequence_align==0.2.0
setuptools==80.9.0
shellingham==1.5.4
six==1.17.0
smart_open==7.5.0
sniffio==1.3.1
soupsieve==2.8.1
sympy==1.14.0
syntok==1.4.4
tenacity==9.1.2
term-image==0.7.2
tinycss2==1.4.0
tinyhost==0.4.18
tokenizers==0.22.2
torch==2.9.1
tornado==6.5.4
tqdm==4.67.1
traitlets==5.14.3
transformers==4.57.3
typer-slim==0.21.1
typing-inspection==0.4.2
typing_extensions==4.15.0
urllib3==2.6.3
wcwidth==0.2.14
webencodings==0.5.1
websockets==15.0.1
Werkzeug==3.1.5
wrapt==2.0.1
zipp==3.23.0
zstandard==0.25.0

Guida per i contributori

Apri la guida per i contributori

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.