From 4a77c965ddb7057081bf204558376f464c4d36d6 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Thu, 11 Dec 2025 21:16:17 +0000
Subject: [PATCH 1/2] python3Packages.optimum-onnx: init at 0.0.2

---
 .../python-modules/optimum-onnx/default.nix   | 67 +++++++++++++++++++
 pkgs/top-level/python-packages.nix            |  2 +
 2 files changed, 69 insertions(+)
 create mode 100644 pkgs/development/python-modules/optimum-onnx/default.nix

diff --git a/pkgs/development/python-modules/optimum-onnx/default.nix b/pkgs/development/python-modules/optimum-onnx/default.nix
new file mode 100644
index 000000000000..f596ed96e0c2
--- /dev/null
+++ b/pkgs/development/python-modules/optimum-onnx/default.nix
@@ -0,0 +1,67 @@
+{
+  lib,
+  buildPythonPackage,
+  fetchFromGitHub,
+
+  # build-system
+  setuptools,
+
+  # dependencies
+  onnx,
+  optimum,
+  transformers,
+
+  # optional-dependencies
+  onnxruntime,
+  # onnxruntime-gpu, unpackaged
+  ruff,
+}:
+
+buildPythonPackage rec {
+  pname = "optimum-onnx";
+  version = "0.0.2";
+  pyproject = true;
+
+  src = fetchFromGitHub {
+    owner = "huggingface";
+    repo = "optimum-onnx";
+    tag = "v${version}";
+    hash = "sha256-wT3yqS64LMuq76Yxs6V6nHfD8vgSfPoJm3hbW7E2zpk=";
+  };
+
+  build-system = [
+    setuptools
+  ];
+
+  pythonRelaxDeps = [
+    "transformers"
+  ];
+  dependencies = [
+    onnx
+    optimum
+    transformers
+  ];
+
+  optional-dependencies = {
+    onnxruntime = [
+      onnxruntime
+    ];
+    # onnxruntime-gpu = [ onnxruntime-gpu ];
+    quality = [
+      ruff
+    ];
+  };
+
+  pythonImportsCheck = [ "optimum.onnxruntime" ];
+
+  # Almost all tests need internet access
+  doCheck = false;
+
+  meta = {
+    description = "Export your model to ONNX and run inference with ONNX Runtime";
+    homepage = "https://github.com/huggingface/optimum-onnx";
+    changelog = "https://github.com/huggingface/optimum-onnx/releases/tag/${src.tag}";
+    license = lib.licenses.asl20;
+    maintainers = with lib.maintainers; [ GaetanLepage ];
+  };
+}
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index 4518f9bb5f45..ed2e88068c74 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -11436,6 +11436,8 @@ self: super: with self; {
 
   optimum = callPackage ../development/python-modules/optimum { };
 
+  optimum-onnx = callPackage ../development/python-modules/optimum-onnx { };
+
   optree = callPackage ../development/python-modules/optree { };
 
   optuna = callPackage ../development/python-modules/optuna { };

From 330489f011c8c447baaf99692ffc83aa252cb6e0 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Thu, 11 Dec 2025 19:42:10 +0000
Subject: [PATCH 2/2] python3Packages.sentence-transformers: 5.1.2 -> 5.2.0

Diff: https://github.com/huggingface/sentence-transformers/compare/v5.1.2...v5.2.0

Changelog: https://github.com/huggingface/sentence-transformers/releases/tag/v5.2.0
---
 .../sentence-transformers/default.nix         | 25 ++++++++++++-------
 1 file changed, 16 insertions(+), 9 deletions(-)

diff --git a/pkgs/development/python-modules/sentence-transformers/default.nix b/pkgs/development/python-modules/sentence-transformers/default.nix
index dde406954171..15d7b590b131 100644
--- a/pkgs/development/python-modules/sentence-transformers/default.nix
+++ b/pkgs/development/python-modules/sentence-transformers/default.nix
@@ -8,11 +8,7 @@
   setuptools,
 
   # dependencies
-  accelerate,
-  datasets,
   huggingface-hub,
-  optimum,
-  pillow,
   scikit-learn,
   scipy,
   torch,
@@ -20,6 +16,15 @@
   transformers,
   typing-extensions,
 
+  # optional-dependencies
+  # image
+  pillow,
+  # train
+  accelerate,
+  datasets,
+  # onnx
+  optimum-onnx,
+
   # tests
   pytestCheckHook,
   pytest-cov-stub,
@@ -27,21 +32,20 @@
 
 buildPythonPackage rec {
   pname = "sentence-transformers";
-  version = "5.1.2";
+  version = "5.2.0";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "huggingface";
     repo = "sentence-transformers";
     tag = "v${version}";
-    hash = "sha256-FNJ4mWBcgy3J8ZJtHt+uBgNmvMnqphj+sLMmBvgdB1k=";
+    hash = "sha256-WD5uTfAbDYYeSXlgznSs4XyN1fAILxILmmSHmLosmV4=";
   };
 
   build-system = [ setuptools ];
 
   dependencies = [
     huggingface-hub
-    pillow
     scikit-learn
     scipy
     torch
@@ -51,12 +55,15 @@ buildPythonPackage rec {
   ];
 
   optional-dependencies = {
+    image = [
+      pillow
+    ];
     train = [
       accelerate
       datasets
     ];
-    onnx = [ optimum ] ++ optimum.optional-dependencies.onnxruntime;
-    # onnx-gpu = [ optimum ] ++ optimum.optional-dependencies.onnxruntime-gpu;
+    onnx = [ optimum-onnx ] ++ optimum-onnx.optional-dependencies.onnxruntime;
+    # onnx-gpu = [ optimum-onnx ] ++ optimum-onnx.optional-dependencies.onnxruntime-gpu;
     # openvino = [ optimum-intel ] ++ optimum-intel.optional-dependencies.openvino;
   };