Set intra_threads=1 for test_transformers_wav2vec2bert as well

3manifold · 3manifold · commit 50a74ceef49a · 2026-02-04T14:10:07.000+01:00
diff --git a/python/tests/test_transformers.py b/python/tests/test_transformers.py
@@ -1091,13 +1091,13 @@ def test_transformers_wav2vec2bert(
         )
 
         device = "cuda" if os.environ.get("CUDA_VISIBLE_DEVICES") else "cpu"
-        cpu_threads = int(os.environ.get("OMP_NUM_THREADS", 0))
+        # cpu_threads = int(os.environ.get("OMP_NUM_THREADS", 0))
         model = ctranslate2.models.Wav2Vec2Bert(
             output_dir,
             device=device,
             device_index=[0],
             compute_type="int8",
-            intra_threads=cpu_threads,
+            intra_threads=1,
             inter_threads=1,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -1091,13 +1091,13 @@ def test_transformers_wav2vec2bert(`
`1091`	`1091`	`)`
`1092`	`1092`
`1093`	`1093`	`device = "cuda" if os.environ.get("CUDA_VISIBLE_DEVICES") else "cpu"`
`1094`		`- cpu_threads = int(os.environ.get("OMP_NUM_THREADS", 0))`
	`1094`	`+ # cpu_threads = int(os.environ.get("OMP_NUM_THREADS", 0))`
`1095`	`1095`	`model = ctranslate2.models.Wav2Vec2Bert(`
`1096`	`1096`	`output_dir,`
`1097`	`1097`	`device=device,`
`1098`	`1098`	`device_index=[0],`
`1099`	`1099`	`compute_type="int8",`
`1100`		`- intra_threads=cpu_threads,`
	`1100`	`+ intra_threads=1,`
`1101`	`1101`	`inter_threads=1,`
`1102`	`1102`	`)`
`1103`	`1103`