python3Packages.timm: fix (#504741)

2026-03-29 21:13:17 +00:00
parent e1fd980298 57afcdcf74
commit 1696905deb
1 changed files with 12 additions and 0 deletions
@@ -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 = [