chore: Disable transformers.generation.utils logging

alexandrainst · Oct 24, 2024 · 3626c63 · 3626c63
1 parent 3050ca4
commit 3626c63
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/coral/utils.py b/src/coral/utils.py
@@ -47,6 +47,7 @@ def block_terminal_output() -> None:
     logging.getLogger("accelerate").setLevel(logging.ERROR)
     logging.getLogger("pyctcdecode").setLevel(logging.ERROR)
     logging.getLogger("transformers.models.whisper").setLevel(logging.ERROR)
+    logging.getLogger("transformers.generation.utils").setLevel(logging.ERROR)
 
 
 class transformers_output_ignored: