Skip to content

Commit 13d8e55

Browse files
authored
Relaxes the validation of model task types during multi-model deployment creation. (#1157)
1 parent 6b5121d commit 13d8e55

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

ads/aqua/model/model.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -304,14 +304,15 @@ def create_multi(
304304
# "Currently only service models are supported for multi model deployment."
305305
# )
306306

307-
if (
308-
source_model.freeform_tags.get(Tags.TASK, UNKNOWN).lower()
309-
not in MultiModelSupportedTaskType
310-
):
311-
raise AquaValueError(
312-
f"Invalid or missing {Tags.TASK} tag for selected model {display_name}. "
313-
f"Currently only `{MultiModelSupportedTaskType.values()}` models are supported for multi model deployment."
314-
)
307+
# TODO uncomment the section below if only the specific types of models should be allowed for multi-model deployment
308+
# if (
309+
# source_model.freeform_tags.get(Tags.TASK, UNKNOWN).lower()
310+
# not in MultiModelSupportedTaskType
311+
# ):
312+
# raise AquaValueError(
313+
# f"Invalid or missing {Tags.TASK} tag for selected model {display_name}. "
314+
# f"Currently only `{MultiModelSupportedTaskType.values()}` models are supported for multi model deployment."
315+
# )
315316

316317
display_name_list.append(display_name)
317318

0 commit comments

Comments
 (0)