Revert "Revoke cpu_threads in test_transformers_wav2vec2 similar to test_transformers_wav2vec2bert"

3manifold · 3manifold · commit f6cea405a5ea · 2026-01-29T10:57:08.000+01:00
This reverts commit f7c71bd.
diff --git a/python/tests/test_transformers.py b/python/tests/test_transformers.py
@@ -1009,13 +1009,13 @@ def test_transformers_wav2vec2(
         )
 
         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.Wav2Vec2(
             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
`@@ -1009,13 +1009,13 @@ def test_transformers_wav2vec2(`
`1009`	`1009`	`)`
`1010`	`1010`
`1011`	`1011`	`device = "cuda" if os.environ.get("CUDA_VISIBLE_DEVICES") else "cpu"`
`1012`		`- cpu_threads = int(os.environ.get("OMP_NUM_THREADS", 0))`
	`1012`	`+ # cpu_threads = int(os.environ.get("OMP_NUM_THREADS", 0))`
`1013`	`1013`	`model = ctranslate2.models.Wav2Vec2(`
`1014`	`1014`	`output_dir,`
`1015`	`1015`	`device=device,`
`1016`	`1016`	`device_index=[0],`
`1017`	`1017`	`compute_type="int8",`
`1018`		`- intra_threads=cpu_threads,`
	`1018`	`+ intra_threads=1,`
`1019`	`1019`	`inter_threads=1,`
`1020`	`1020`	`)`
`1021`	`1021`