From 1f19c7fef715eb44068a45f740d6e5727c8a04c7 Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Fri, 19 Jul 2024 15:01:34 -0500 Subject: [PATCH] Don't pass benchmark as an export arg. --- models/turbine_models/custom_models/pipeline_base.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/models/turbine_models/custom_models/pipeline_base.py b/models/turbine_models/custom_models/pipeline_base.py index 22b44e4e0..b54b158c2 100644 --- 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(