From 16ff5a23cbca991058176aaf2012d4f5684ab2a5 Mon Sep 17 00:00:00 2001 From: Copilot <198982749+Copilot@users.noreply.github.com> Date: Thu, 21 May 2026 17:26:29 +0800 Subject: [PATCH] [fix] use getattr for profiler attrs to support MCA TrainingArguments (#10506) Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com> Co-authored-by: hiyouga <16256802+hiyouga@users.noreply.github.com> --- src/llamafactory/train/tuner.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/llamafactory/train/tuner.py b/src/llamafactory/train/tuner.py index dcde974b7..2079d840c 100644 --- a/src/llamafactory/train/tuner.py +++ b/src/llamafactory/train/tuner.py @@ -80,10 +80,10 @@ def _training_function(config: dict[str, Any]) -> None: if finetuning_args.early_stopping_steps is not None: callbacks.append(EarlyStoppingCallback(early_stopping_patience=finetuning_args.early_stopping_steps)) - if training_args.enable_torch_profiler: + if getattr(training_args, "enable_torch_profiler", False): callbacks.append(TorchProfilerCallback(training_args)) - if training_args.profile_modules: + if getattr(training_args, "profile_modules", None): callbacks.append(ModuleProfilerCallback(training_args.profile_modules)) callbacks.append(ReporterCallback(model_args, data_args, finetuning_args, generating_args)) # add to last