From 73adbc98b2c81577bc1d992ed2e3546bb7d959cf Mon Sep 17 00:00:00 2001
From: rahul-tuli <rahul@neuralmagic.com>
Date: Thu, 30 Nov 2023 17:01:52 +0000
Subject: [PATCH 1/2] [Fix] Append date to major_minor version only when
 building a nightly release

---
 setup.py | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/setup.py b/setup.py
index 9409b07f..f0e03c97 100644
--- a/setup.py
+++ b/setup.py
@@ -26,7 +26,9 @@
 # load and overwrite version and release info from sparseml package
 exec(open(os.path.join("src", "sparsify", "version.py")).read())
 print(f"loaded version {version} from src/sparsify/version.py")
-version_nm_deps = f"{version_major_minor}.0.202308"
+version_nm_deps = (
+    version_major_minor if is_release else f"{version_major_minor}.0.202308"
+)
 
 _PACKAGE_NAME = "sparsify" if is_release else "sparsify-nightly"
 
@@ -37,14 +39,13 @@
     "tensorboard>=2.0.0",
     "setuptools>=56.0.0",
     "optuna>=3.0.2",
-    "onnxruntime-gpu",
-    f"{'sparsezoo' if is_release else 'sparsezoo-nightly'}~={version_nm_deps}",
-    f"{'deepsparse' if is_release else 'deepsparse-nightly'}~={version_nm_deps}",
-    f"{'sparseml' if is_release else 'sparseml-nightly'}[torchvision,yolov5]~={version_nm_deps}",  # noqa E501
+    "onnxruntime",
 ]
 
 _nm_deps = [
-    f"{'sparseml' if is_release else 'sparseml-nightly'}[transformers]~={version_nm_deps}",  # noqa E501
+    f"{'sparsezoo' if is_release else 'sparsezoo-nightly'}~={version_nm_deps}",
+    f"{'deepsparse' if is_release else 'deepsparse-nightly'}~={version_nm_deps}",
+    f"{'sparseml' if is_release else 'sparseml-nightly'}[torchvision,yolov5]~={version_nm_deps}",  # noqa E501
 ]
 
 _dev_deps = [
@@ -73,7 +74,7 @@ def _setup_package_dir() -> Dict:
 
 
 def _setup_install_requires() -> List:
-    return _deps
+    return _nm_deps + _deps
 
 
 def _setup_extras() -> Dict:

From d27e941248957e598948062eb19085c0070533a9 Mon Sep 17 00:00:00 2001
From: rahul-tuli <rahul@neuralmagic.com>
Date: Thu, 30 Nov 2023 17:05:40 +0000
Subject: [PATCH 2/2] missed onnxruntime-gpu

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index f0e03c97..a57ca380 100644
--- a/setup.py
+++ b/setup.py
@@ -39,7 +39,7 @@
     "tensorboard>=2.0.0",
     "setuptools>=56.0.0",
     "optuna>=3.0.2",
-    "onnxruntime",
+    "onnxruntime-gpu",
 ]
 
 _nm_deps = [