diff --git a/flake.lock b/flake.lock
index ba3a71733ee..d64b76b1fca 100644
--- a/flake.lock
+++ b/flake.lock
@@ -978,16 +978,15 @@
         "nixpkgs": "nixpkgs_6"
       },
       "locked": {
-        "lastModified": 1738229197,
-        "narHash": "sha256-K/YJSFhzP0vN23GMfM1HVMtSzaM488hh12ggsMtKMG0=",
+        "lastModified": 1738251792,
+        "narHash": "sha256-pHwXzy/TmIAKi1d9qs5USwxDtYf4DucGK9mhGmXxUfQ=",
         "owner": "huggingface",
         "repo": "text-generation-inference-nix",
-        "rev": "cfcddaf3044f59c3fbd335935ac3c0e9f458d824",
+        "rev": "c7f53c9ba51f7d1502a0f1c64bf81aad57f2a097",
         "type": "github"
       },
       "original": {
         "owner": "huggingface",
-        "ref": "moe_0_8_1",
         "repo": "text-generation-inference-nix",
         "type": "github"
       }
diff --git a/flake.nix b/flake.nix
index 6c9f50149ae..83cedfa620f 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/moe_0_8_1";
+    tgi-nix.url = "github:huggingface/text-generation-inference-nix";
     nixpkgs.follows = "tgi-nix/nixpkgs";
     flake-utils.url = "github:numtide/flake-utils";
     rust-overlay = {