Unverified Commit 1696905d authored by kirillrdy's avatar kirillrdy Committed by GitHub
Browse files

python3Packages.timm: fix (#504741)

parents e1fd9802 57afcdcf

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 = [