Closes caikit#134

Signed-off-by: Joe Olson <[email protected]>
opendatahub-io · Aug 21, 2023 · 4762bf5 · 4762bf5
1 parent 2c2efc3
commit 4762bf5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/examples/run_fine_tuning.py b/examples/run_fine_tuning.py
@@ -313,7 +313,11 @@ 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)
+ base_model = model_type.bootstrap(
+ args.model_name, 
+ tokenizer_name=args.model_name, 
+ torch_dtype=args.torch_dtype
+ )
 
  # Then actually train the model & save it
  print_colored("[Starting the training...]")