diff --git a/requirements/requirements.cpu.txt b/requirements/requirements.cpu.txt index 47f85bb5b..548e13b3d 100644 --- a/requirements/requirements.cpu.txt +++ b/requirements/requirements.cpu.txt @@ -1,2 +1,2 @@ -onnxruntime<=1.15.1 +onnxruntime<=1.17.0 GPUtil==1.4.0 \ No newline at end of file diff --git a/requirements/requirements.gpu.txt b/requirements/requirements.gpu.txt index 4347b46d4..2d42277b9 100644 --- a/requirements/requirements.gpu.txt +++ b/requirements/requirements.gpu.txt @@ -1 +1 @@ -onnxruntime-gpu<=1.15.1 \ No newline at end of file +onnxruntime-gpu<=1.17.0 \ No newline at end of file diff --git a/requirements/requirements.vino.txt b/requirements/requirements.vino.txt index b16b5922e..3233bc073 100644 --- a/requirements/requirements.vino.txt +++ b/requirements/requirements.vino.txt @@ -1 +1 @@ -onnxruntime-openvino<=1.15.1 \ No newline at end of file +onnxruntime-openvino<=1.17.0 \ No newline at end of file diff --git a/setup.py b/setup.py index f14488f31..130f4911d 100644 --- a/setup.py +++ b/setup.py @@ -17,7 +17,8 @@ def read_requirements(path): setuptools.setup( - name="inference-development", + name="inference-gpu", + version="0.20.1", author="Roboflow", author_email="help@roboflow.com", description="With no prior knowledge of machine learning or device-specific deployment, you can deploy a computer vision model to a range of devices and environments using Roboflow Inference.", @@ -44,23 +45,23 @@ def read_requirements(path): install_requires=read_requirements( [ "requirements/_requirements.txt", - "requirements/requirements.cpu.txt", + "requirements/requirements.gpu.txt", "requirements/requirements.cli.txt", - "requirements/requirements.clip.txt", - "requirements/requirements.http.txt", "requirements/requirements.sdk.http.txt", - "requirements/requirements.gaze.txt", - "requirements/requirements.groundingdino.txt", - "requirements/requirements.hosted.txt", - "requirements/requirements.waf.txt", - "requirements/requirements.yolo_world.txt", - "requirements/requirements.code_analysis.txt", - "requirements/requirements.test.unit.txt", - "requirements/requirements.test.integration.txt", ] ), extras_require={ + "clip": read_requirements("requirements/requirements.clip.txt"), + "gaze": read_requirements("requirements/requirements.gaze.txt"), + "grounding-dino": read_requirements( + "requirements/requirements.groundingdino.txt" + ), + "hosted": read_requirements("requirements/requirements.hosted.txt"), + "http": read_requirements("requirements/requirements.http.txt"), "sam": read_requirements("requirements/requirements.sam.txt"), + "waf": read_requirements("requirements/requirements.waf.txt"), + "yolo-world": read_requirements("requirements/requirements.yolo_world.txt"), + "transformers": read_requirements("requirements/requirements.transformers.txt"), }, classifiers=[ "Programming Language :: Python :: 3",