python3Packages.x-transformers: limit openmp threads during tests

2026-06-25 00:44:12 +02:00
parent b153953686
commit b6e39b216d
1 changed files with 4 additions and 0 deletions
@@ -42,6 +42,10 @@ buildPythonPackage (finalAttrs: {

  nativeCheckInputs = [ pytestCheckHook ];

+  preCheck = ''
+    export OMP_NUM_THREADS=1
+  '';
+
  # RuntimeError: torch.compile is not supported on Python 3.14+
  disabledTests = lib.optionals (pythonAtLeast "3.14") [ "test_up" ];