chore: Set OMP_NUM_THREADS=4

alexandrainst · Oct 23, 2024 · 7a0760a · 7a0760a
1 parent c82a014
commit 7a0760a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/makefile b/makefile
@@ -136,7 +136,7 @@ roest-315m:  ## Train the Røst-315M model
 		per_device_batch_size=64
 
 roest-809m:  ## Train the Røst-809M model
-	@accelerate launch \
+	@OMP_NUM_THREADS=4 accelerate launch \
 		--use-deepspeed \
 		src/scripts/finetune_asr_model.py \
 		model=whisper-large-turbo \