Don't pass benchmark as an export arg.

nod-ai · Jul 19, 2024 · 1f19c7f · 1f19c7f
1 parent bbcc424
commit 1f19c7f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/models/turbine_models/custom_models/pipeline_base.py b/models/turbine_models/custom_models/pipeline_base.py
@@ -391,7 +391,8 @@ def __init__(
                 )
         for submodel in self.map.keys():
             for key, value in map_arguments.items():
-                self.map = merge_export_arg(self.map, value, key)
+                if key != "benchmark":
+                    self.map = merge_export_arg(self.map, value, key)
             for key, value in self.map[submodel].get("export_args", {}).items():
                 if key == "hf_model_name":
                     self.map[submodel]["keywords"].append(