Update onnx/builder.py

onnx/builder.py

@@ -92,7 +92,7 @@ def build_vision(args):
         "--output_model", fpath_4,
         "--block_size", str(32),
     ]
-    if args.precision == "fp32": cmd.extend(["--accuracy_level", str(4)])
+    if args.precision == torch.float32: cmd.extend(["--accuracy_level", str(4)])
     subprocess.run(cmd)
     shutil.rmtree(temp_folder_3)
 
@@ -182,7 +182,7 @@ def build_text(args):
         "exclude_embeds": "true",
         "filename": "phi-3-v-128k-instruct-text.onnx",
     }
-    if args.precision == "fp32": extra_options["int4_accuracy_level"] = 4
+    if args.precision == torch.float32: extra_options["int4_accuracy_level"] = 4
     create_model(model_name, args.input, args.output, precision, args.execution_provider, args.cache_dir, **extra_options)
 
 def get_args():