From 62fb552669b15e38eca6d78871c4edc6a2f2e870 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Sun, 23 Mar 2025 21:49:09 +0100
Subject: [PATCH] Torch 2.6

---
 flake.lock | 8 ++++----
 flake.nix  | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/flake.lock b/flake.lock
index d049a71d..ab6298de 100644
--- a/flake.lock
+++ b/flake.lock
@@ -978,16 +978,16 @@
         "nixpkgs": "nixpkgs_6"
       },
       "locked": {
-        "lastModified": 1741617161,
-        "narHash": "sha256-cwKYAsIVSLtoLbG48+oi3NkSrvuZRLYs8lkJmpDsTw0=",
+        "lastModified": 1742579922,
+        "narHash": "sha256-xpkilkp+9zWwdFkn7Dv2ggdEulZf8amSNdh6bvJMJiw=",
         "owner": "huggingface",
         "repo": "text-generation-inference-nix",
-        "rev": "5946021ec6cb6aae18158a9dc27f893cfbab2925",
+        "rev": "557154f30e4cc3c7641509bdd45a3cdf15f843e9",
         "type": "github"
       },
       "original": {
         "owner": "huggingface",
-        "ref": "kernels-0.2.0",
+        "ref": "torch_2.6",
         "repo": "text-generation-inference-nix",
         "type": "github"
       }
diff --git a/flake.nix b/flake.nix
index 7ddd3b92..84c95964 100644
--- a/flake.nix
+++ b/flake.nix
@@ -5,7 +5,7 @@
       inputs.nixpkgs.follows = "tgi-nix/nixpkgs";
     };
     nix-filter.url = "github:numtide/nix-filter";
-    tgi-nix.url = "github:huggingface/text-generation-inference-nix/kernels-0.2.0";
+    tgi-nix.url = "github:huggingface/text-generation-inference-nix/torch_2.6";
     nixpkgs.follows = "tgi-nix/nixpkgs";
     flake-utils.url = "github:numtide/flake-utils";
     rust-overlay = {