Fix flake input

flake.lock

@@ -43,15 +43,15 @@
       "locked": {
         "lastModified": 1744736115,
         "narHash": "sha256-9PPp6XHoMx9jZjwCP7XvAlc52+TmmVuCbUqwh3snuI8=",
-        "ref": "refs/heads/main",
+        "owner": "huggingface",
+        "repo": "kernel-builder",
         "rev": "319af881b27c3645dfc33128f99092c7c1176281",
-        "revCount": 137,
-        "type": "git",
-        "url": "ssh://[email protected]/huggingface/kernel-builder"
+        "type": "github"
       },
       "original": {
-        "type": "git",
-        "url": "ssh://[email protected]/huggingface/kernel-builder"
+        "owner": "huggingface",
+        "repo": "kernel-builder",
+        "type": "github"
       }
     },
     "nixpkgs": {

flake.nix

@@ -2,7 +2,7 @@
   description = "Flake for FlashMLA kernel";
 
   inputs = {
-    kernel-builder.url = "git+ssh://git@github.com/huggingface/kernel-builder";
+    kernel-builder.url = "github:huggingface/kernel-builder";
   };
 
   outputs =