File tree Expand file tree Collapse file tree 1 file changed +9
-8
lines changed Expand file tree Collapse file tree 1 file changed +9
-8
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments