diff --git a/nix-builder/overlay.nix b/nix-builder/overlay.nix index 4c451919..3a54d39c 100644 --- a/nix-builder/overlay.nix +++ b/nix-builder/overlay.nix @@ -77,18 +77,20 @@ in else python-self.callPackage ./pkgs/python-modules/cuda-python { }; - huggingface-hub = python-super.huggingface-hub.overridePythonAttrs (old: rec { - version = "1.3.0"; + huggingface-hub = python-super.huggingface-hub.overridePythonAttrs (prevAttrs: rec { + version = "1.8.0"; src = python-super.fetchPypi { pname = "huggingface_hub"; inherit version; - hash = "sha256-KJ4qNYb98B41iClE6qBvvVdDbeJLbmU9H6skhYSs1ms="; + hash = "sha256-xWJ7L9Uh4Ayvjv9KyWW6mI6nUWf61+5y4X+bcYPsY/M="; }; - dependencies = (old.dependencies or [ ]) ++ [ - python-self.httpx - python-self.shellingham - python-self.typer-slim - ]; + dependencies = + (prevAttrs.dependencies or [ ]) + ++ (with python-self; [ + httpx + shellingham + typer + ]); # Skip tests since they require network access. doCheck = false; }); @@ -100,6 +102,27 @@ in } ); + hf-xet = python-super.hf-xet.overridePythonAttrs (prevAttrs: rec { + version = "1.4.2"; + src = final.fetchFromGitHub { + owner = "huggingface"; + repo = "xet-core"; + tag = "v${version}"; + hash = "sha256-UdHEpJztlVI8LPs8Ne9sKe1Nv3kVVk4YLxQ3W8sUPbQ="; + }; + cargoDeps = final.rustPlatform.fetchCargoVendor { + inherit (prevAttrs) + pname + sourceRoot + ; + inherit + version + src + ; + hash = "sha256-GV+XY5uV57yQWVGdRLpGU3eD8Gz2gy6p7OHlF+mlJI4="; + }; + }); + nvidia-cutlass-dsl = python-self.callPackage ./pkgs/python-modules/nvidia-cutlass-dsl { }; nvidia-cutlass-dsl-libs = python-self.callPackage ./pkgs/python-modules/nvidia-cutlass-dsl-libs { }; @@ -135,6 +158,19 @@ in xpuPackages = final.xpuPackages_2025_3; }; + transformers = python-super.transformers.overridePythonAttrs (prevAttrs: rec { + version = "5.3.0"; + src = python-super.fetchPypi { + pname = "transformers"; + inherit version; + hash = "sha256-AJVVs2QCnanilG1B8cXenxXmsd9GsYm3KT8zoWG5xVc="; + }; + + dependencies = (prevAttrs.dependencies or [ ]) ++ [ + python-self.typer + ]; + }); + triton-xpu_2_9 = callPackage ./pkgs/python-modules/triton-xpu { torchVersion = "2.9"; xpuPackages = final.xpuPackages_2025_2;