suppress HF warnings and lower during detector model instantiation (#807

)
NVIDIA · Jul 30, 2024 · f2e5182 · f2e5182
1 parent 8c9b659
commit f2e5182
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/garak/detectors/base.py b/garak/detectors/base.py
@@ -74,6 +74,7 @@ def __init__(self, model_path, target_class, config_root=_config):
             AutoTokenizer,
             TextClassificationPipeline,
             set_seed,
+            logging as transformers_logging,
         )
 
         if _config.run.seed is not None:
@@ -85,6 +86,9 @@ def __init__(self, model_path, target_class, config_root=_config):
         self.detector_model_path = model_path
         self.detector_target_class = target_class
 
+        orig_loglevel = transformers_logging.get_verbosity()
+        transformers_logging.set_verbosity_error()
+
         self.detector_model = AutoModelForSequenceClassification.from_pretrained(
             self.detector_model_path
         )
@@ -95,6 +99,8 @@ def __init__(self, model_path, target_class, config_root=_config):
             model=self.detector_model, tokenizer=self.detector_tokenizer
         )
 
+        transformers_logging.set_verbosity(orig_loglevel)
+
         self.graceful_fail = False
 
     def detect(self, attempt: garak.attempt.Attempt) -> List[float]: