From a6d09f523bf9241480e9d86f41c2a0ad4db880d2 Mon Sep 17 00:00:00 2001
From: Tom Aarsen <Cubiegamedev@gmail.com>
Date: Thu, 25 Apr 2024 11:45:44 +0200
Subject: [PATCH] Add info-level logger message

---
 src/transformers/integrations/integration_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/transformers/integrations/integration_utils.py b/src/transformers/integrations/integration_utils.py
index cacf3e7766508f..2839ee876ed456 100644
--- a/src/transformers/integrations/integration_utils.py
+++ b/src/transformers/integrations/integration_utils.py
@@ -789,7 +789,7 @@ def setup(self, args, state, model, **kwargs):
             try:
                 self._wandb.config["model/num_parameters"] = model.num_parameters()
             except AttributeError:
-                pass
+                logger.info("Could not log the number of model parameters in Weights & Biases.")
 
             # log the initial model and architecture to an artifact
             with tempfile.TemporaryDirectory() as temp_dir: