Set intra_threads=1 for test_transformers_wav2vec2bert as well

3manifold · 3manifold · commit 6f37ce34dc63 · 2026-01-29T10:58:11.000+01:00
diff --git a/python/tests/test_transformers.py b/python/tests/test_transformers.py
@@ -1089,13 +1089,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
`@@ -1089,13 +1089,13 @@ def test_transformers_wav2vec2bert(`
`1089`	`1089`	`)`
`1090`	`1090`
`1091`	`1091`	`device = "cuda" if os.environ.get("CUDA_VISIBLE_DEVICES") else "cpu"`
`1092`		`- cpu_threads = int(os.environ.get("OMP_NUM_THREADS", 0))`
	`1092`	`+ # cpu_threads = int(os.environ.get("OMP_NUM_THREADS", 0))`
`1093`	`1093`	`model = ctranslate2.models.Wav2Vec2Bert(`
`1094`	`1094`	`output_dir,`
`1095`	`1095`	`device=device,`
`1096`	`1096`	`device_index=[0],`
`1097`	`1097`	`compute_type="int8",`
`1098`		`- intra_threads=cpu_threads,`
	`1098`	`+ intra_threads=1,`
`1099`	`1099`	`inter_threads=1,`
`1100`	`1100`	`)`
`1101`	`1101`