Remove log message in DiT

AI-Riksarkivet · May 7, 2024 · a873b80 · a873b80
1 parent 92fcc48
commit a873b80
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/htrflow_core/models/huggingface/dit.py b/src/htrflow_core/models/huggingface/dit.py
@@ -78,6 +78,4 @@ def _label_based_on_return_format(self, logits):
         else:
             probabilities = torch.nn.functional.softmax(logits, dim=-1)
             label_ = {self.model.config.id2label[id]: prob.item() for id, prob in enumerate(probabilities)}
-
-        logger.info(f"Prediction complete. Return format: {self.return_format},  label: {label_}.")
         return label_