From d0147b598c82a94924ee29397dddd1725f4b837b Mon Sep 17 00:00:00 2001 From: surya10602 Date: Fri, 6 Feb 2026 02:03:07 +0530 Subject: [PATCH] feat(integrations): Add support for id and resume args in SwanLabCallback --- src/transformers/integrations/integration_utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/transformers/integrations/integration_utils.py b/src/transformers/integrations/integration_utils.py index aabfd0bbe268..b14259ca443a 100755 --- a/src/transformers/integrations/integration_utils.py +++ b/src/transformers/integrations/integration_utils.py @@ -2227,7 +2227,7 @@ class SwanLabCallback(TrainerCallback): A [`TrainerCallback`] that logs metrics, media, model checkpoints to [SwanLab](https://swanlab.cn/). """ - def __init__(self): + def __init__(self, **kwargs): if not is_swanlab_available(): raise RuntimeError("SwanLabCallback requires swanlab to be installed. Run `pip install swanlab`.") import swanlab @@ -2235,6 +2235,7 @@ def __init__(self): self._swanlab = swanlab self._initialized = False self._log_model = os.getenv("SWANLAB_LOG_MODEL", None) + self._init_kwargs = kwargs def setup(self, args, state, model, **kwargs): """ @@ -2302,6 +2303,7 @@ def setup(self, args, state, model, **kwargs): init_args["project"] = os.getenv("SWANLAB_PROJECT", None) if self._swanlab.get_run() is None: + init_args.update(self._init_kwargs) self._swanlab.init( **init_args, )