diff --git a/model/model_training/tools/export_model.py b/model/model_training/tools/export_model.py index 8ce456085e..b0959f862d 100644 --- a/model/model_training/tools/export_model.py +++ b/model/model_training/tools/export_model.py @@ -41,7 +41,7 @@ def main(): elif args.dtype in ("bfloat16", "bf16"): torch_dtype = torch.bfloat16 else: - print(f"Unsupported dtpye: {args.dtype}") + print(f"Unsupported dtype: {args.dtype}") sys.exit(1) if not args.hf_repo_name and not args.output_folder: