diff --git a/examples/run_fine_tuning.py b/examples/run_fine_tuning.py index ce11a5af..9d7d5bf4 100644 --- a/examples/run_fine_tuning.py +++ b/examples/run_fine_tuning.py @@ -314,9 +314,7 @@ def export_model_preds(preds_file, predictions, validation_stream): # Init the resource & Build the tuning config from our dataset/arg info print_colored("[Loading the base model resource...]") base_model = model_type.bootstrap( - args.model_name, - tokenizer_name=args.model_name, - torch_dtype=args.torch_dtype + args.model_name, tokenizer_name=args.model_name, torch_dtype=args.torch_dtype ) # Then actually train the model & save it