[Fix] Fix test file error (#40973)

Fix test file error
This commit is contained in:
YangKai0616
2025-10-09 23:30:53 +08:00
committed by GitHub
parent 34b861abd1
commit 26b5b52676

View File

@ -5907,7 +5907,7 @@ if is_torch_available():
)
optim_test_params.append(
(
TrainingArguments(optim=OptimizerNames.ADAMW_TORCH_8BIT, output_dir="None"),
OptimizerNames.ADAMW_TORCH_8BIT,
AdamW8bit,
default_adam_kwargs,
)