diff --git a/onnx_scripts/convert_to_onnx.py b/onnx_scripts/convert_to_onnx.py index 0458b4f..8fb17d4 100644 --- a/onnx_scripts/convert_to_onnx.py +++ b/onnx_scripts/convert_to_onnx.py @@ -80,6 +80,6 @@ if __name__ == "__main__": ckpt_path=os.path.join("checkpoint_files", model_checkpoint), fp16=False ) - output_file = "convert_to_onnx.py" + ".onnx" + output_file = "convert_to_onnx" + ".onnx" print("Conversion complete stored at: ", os.path.join("onnx_scripts", output_file))