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(
304
304
# "Currently only service models are supported for multi model deployment."
305
305
# )
306
306
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
+ # )
315
316
316
317
display_name_list .append (display_name )
317
318
You can’t perform that action at this time.
0 commit comments