Commit d842a192 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python312Packages.transformers: fix darwin by skipping torch.distributed import

parent 386ec507

pkgs/development/python-modules/transformers/default.nix

+10 −0
Original line number Diff line number Diff line 
{

  lib,

  stdenv,

  buildPythonPackage,

  fetchFromGitHub,
@@ -68,6 +69,15 @@ buildPythonPackage rec { 
    hash = "sha256-TQQ+w+EH/KWLE7iaaAHGxfE74hCiLXcqlIr1TIBFGvo=";

  };



  # torch.distributed is not available on darwin

  # Fix submitted upstream in https://github.com/huggingface/transformers/pull/35133

  postPatch = lib.optionalString stdenv.hostPlatform.isDarwin ''

    substituteInPlace src/transformers/pytorch_utils.py \

      --replace-fail \

        'if is_torch_greater_or_equal("2.5"):' \

        'if False:'

  '';



  build-system = [ setuptools ];



  dependencies = [