olmocr/setup.py

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/"],
)
actually try setup.py 2025-03-13 13:51:59 -07:00			`from setuptools import setup`

			`setup(`
move the gpu reqs to setup.py altogether 2025-03-13 13:53:59 -07:00			`extras_require={`
			`"gpu": [`
			`"sgl-kernel==0.0.3.post1",`
			`"sglang[all]==0.4.2",`
			`"flashinfer"`
			`],`
			`},`
actually try setup.py 2025-03-13 13:51:59 -07:00			`dependency_links=["https://flashinfer.ai/whl/cu124/torch2.4/flashinfer/"],`
			`)`