fix (serve): added dict config handle in serve model postprocess func

moverseai · Oct 9, 2024 · ffed4cf · ffed4cf
1 parent 330c746
commit ffed4cf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/moai/serve/model.py b/moai/serve/model.py
@@ -256,6 +256,8 @@ def postprocess(
         log.debug(f"Postprocessing outputs:\n{data['__moai__']}")
         outs = []  # TODO: corner case with no postproc crashes, fix it
         for k, p in self.postproc.items():
+            if isinstance(p, omegaconf.dictconfig.DictConfig):
+                continue
             res = p(data, data["__moai__"]["json"])
             if len(outs) == 0:
                 outs = res