fix examples in docs

Author: apoorvkh

scripts/examples/accelerate_train.py

@@ -117,7 +117,7 @@ def main(
     train_dataset = load_training_data(tokenizer_name=model_config.name, dataset_config=dataset_config)
 
     # Launch training
-    results = launcher.run(train, (model, train_dataset, batch_size, output_dir))
+    results = launcher.run(train, model, train_dataset, batch_size, output_dir)
 
     # Loading trained model from checkpoint
     checkpoint_path = results.rank(0)

scripts/examples/deepspeed_train.py

@@ -111,7 +111,7 @@ def main(
     train_dataset = load_training_data(tokenizer_name=model_name, dataset_config=dataset_config)
 
     # Launch training
-    launcher.run(train, (model, train_dataset, str(deepspeed_config), str(checkpoint_dir)))
+    launcher.run(train, model, train_dataset, str(deepspeed_config), str(checkpoint_dir))
 
     # Loading trained model from checkpoint
     state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir)

scripts/examples/lightning_train.py

@@ -126,7 +126,7 @@ def main(
     )
 
     # Launch training
-    results = launcher.run(train, (model, train_dataset))
+    results = launcher.run(train, model, train_dataset)
 
     # Loading trained model from checkpoint
     checkpoint_path = results.rank(0)

scripts/examples/transformers_train.py

@@ -102,7 +102,7 @@ def main(
     )
 
     # Launch training
-    results = launcher.run(train, (model, train_dataset, training_args))
+    results = launcher.run(train, model, train_dataset, training_args)
 
     # Loading trained model from checkpoint
     checkpoint_path = results.rank(0)