stringify config (huggingface#25637)

* stringify config * apply code formatting
parambharat · Sep 26, 2023 · a4ee6db · a4ee6db
1 parent 0f32d69
commit a4ee6db
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/transformers/integrations.py b/src/transformers/integrations.py
@@ -1283,8 +1283,12 @@ def _log_trainer_parameters(self, args):
         self._metadata_namespace[NeptuneCallback.trainer_parameters_key] = args.to_sanitized_dict()
 
     def _log_model_parameters(self, model):
+        from neptune.utils import stringify_unsupported
+
         if model and hasattr(model, "config") and model.config is not None:
-            self._metadata_namespace[NeptuneCallback.model_parameters_key] = model.config.to_dict()
+            self._metadata_namespace[NeptuneCallback.model_parameters_key] = stringify_unsupported(
+                model.config.to_dict()
+            )
 
     def _log_hyper_param_search_parameters(self, state):
         if state and hasattr(state, "trial_name"):