Fix Roboflow node: handle onnxruntime-gpu 1.17.0

roboflow · Sep 24, 2024 · 9d3bb43 · 9d3bb43
1 parent e1f1c0c
commit 9d3bb43
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 15 deletions.
diff --git 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
diff --git a/requirements/requirements.gpu.txt b/requirements/requirements.gpu.txt
@@ -1 +1 @@
-onnxruntime-gpu<=1.15.1
+onnxruntime-gpu<=1.17.0
diff --git a/requirements/requirements.vino.txt b/requirements/requirements.vino.txt
@@ -1 +1 @@
-onnxruntime-openvino<=1.15.1
+onnxruntime-openvino<=1.17.0
diff --git 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="[email protected]",
     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",
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		onnxruntime-openvino<=1.15.1
		onnxruntime-openvino<=1.17.0