Enabled non-dual conversion by default (better results)

appliedAI-Initiative · Sep 5, 2023 · 57fbc1f · 57fbc1f
1 parent 90b666b
commit 57fbc1f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/autodev/run_onnx_conversion.py b/autodev/run_onnx_conversion.py
@@ -21,7 +21,7 @@
     tokenizer = model_factory.create_tokenizer()
 
     # apply conversion
-    #ONNXConversion().convert(model, tokenizer, Path("models/santacoder_onnx"))
-    ONNXConversion().convert_santacoder_dual_models(model, tokenizer, Path("models/santacoder_onnx_cached"))
+    ONNXConversion().convert(model, tokenizer, Path("models/santacoder_onnx"))
+    #ONNXConversion().convert_santacoder_dual_models(model, tokenizer, Path("models/santacoder_onnx_cached"))
 
     log.info("Done")