mirror of
https://github.com/allenai/olmocr.git
synced 2025-08-15 04:11:59 +00:00
13 lines
272 B
Python
13 lines
272 B
Python
from setuptools import setup
|
|
|
|
setup(
|
|
extras_require={
|
|
"gpu": [
|
|
"sgl-kernel==0.0.3.post1",
|
|
"sglang[all]==0.4.2",
|
|
"flashinfer"
|
|
],
|
|
},
|
|
dependency_links=["https://flashinfer.ai/whl/cu124/torch2.4/flashinfer/"],
|
|
)
|