dev(narugo): add wd14 syncer, ci skip

deepghs · May 12, 2024 · f8d873b · f8d873b
1 parent 859e829
commit f8d873b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/zoo/wd14/sync.py b/zoo/wd14/sync.py
@@ -15,6 +15,7 @@
 from tqdm import tqdm
 
 from imgutils.tagging.wd14 import MODEL_NAMES
+from imgutils.utils import open_onnx_model
 
 logging.try_init_root(logging.INFO)
 
@@ -104,7 +105,7 @@ def _is_fc(name):
             onnx.save_model(model, onnx_file)
 
             logging.info(f'Loading and testing for the exported model {onnx_file!r}.')
-            session = onnxruntime.InferenceSession(onnx_file)
+            session = open_onnx_model(onnx_file)
             assert len(session.get_inputs()) == 1
             assert len(session.get_outputs()) == 2
             assert session.get_outputs()[1].name == embs_outputs[0]