From c6ad74b5df063841f65313c31c17cdffe4c37987 Mon Sep 17 00:00:00 2001 From: Liyu Xiao Date: Thu, 22 May 2025 15:04:19 -0400 Subject: [PATCH] reverted .onnx to .cptx --- convert_to_onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/convert_to_onnx.py b/convert_to_onnx.py index a6b6ca7..4616290 100644 --- a/convert_to_onnx.py +++ b/convert_to_onnx.py @@ -72,7 +72,7 @@ def convert_to_onnx(ckpt_path, fp16=False): if __name__ == "__main__": from checkpoint_files import CHECKPOINTS_DIR - model_checkpoint = "inference_recognition_model.onnx" + model_checkpoint = "inference_recognition_model.ckpt" print("Converting to ONNX...")