Commit 57afcdcf authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python3Packages.timm: fix

parent 2946b531

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

+12 −0
Original line number Diff line number Diff line
@@ -33,6 +33,18 @@ buildPythonPackage (finalAttrs: { 
    hash = "sha256-pbzDoNRRwz41b4X40yBp7oTcJ2e/Y2dKyj9XbEX5c34=";

  };



  # Fix torch 2.11.0 compatibility

  # AttributeError: 'AdamWLegacy' object has no attribute '_cuda_graph_capture_health_check'

  postPatch = ''

    substituteInPlace \

      timm/optim/adopt.py \

      timm/optim/adamw.py \

      timm/optim/nadamw.py \

      --replace-fail \

        "_cuda_graph_capture_health_check" \

        "_accelerator_graph_capture_health_check"

  '';



  build-system = [ pdm-backend ];



  dependencies = [