Remove deprecated training arguments (#36946)

* Remove deprecated training arguments * More fixes * More fixes * More fixes
2025-03-27 00:44:48 +08:00
parent 44715225e3
commit 2b550c47b2
8 changed files with 15 additions and 108 deletions
--- a/tests/trainer/test_trainer_distributed.py
+++ b/tests/trainer/test_trainer_distributed.py
@@ -200,6 +200,8 @@ if __name__ == "__main__":
    model = RegressionModel()
    training_args.per_device_train_batch_size = 1
    training_args.max_steps = 1
-    training_args.dispatch_batches = False
+    training_args.accelerator_config = {
+        "dispatch_batches": False,
+    }
    trainer = Trainer(model, training_args, train_dataset=train_dataset)
    trainer.train()