Closes caikit#134

Signed-off-by: Joe Olson <[email protected]>
opendatahub-io · Aug 21, 2023 · 19afcaf · 19afcaf
1 parent 4762bf5
commit 19afcaf
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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