Skip to content

Commit d6b66f1

Browse files
committed
updated error messages
1 parent 44c25b8 commit d6b66f1

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

ads/aqua/model/model.py

+5-7
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,7 @@
8181
ImportModelDetails,
8282
ModelValidationResult,
8383
)
84+
from ads.aqua.model.enums import MultiModelSupportedTaskType
8485
from ads.common.auth import default_signer
8586
from ads.common.oci_resource import SEARCH_TYPE, OCIResource
8687
from ads.common.utils import (
@@ -184,12 +185,8 @@ def create(
184185
target_project = project_id or PROJECT_OCID
185186
target_compartment = compartment_id or COMPARTMENT_OCID
186187

187-
# Skip model copying if it is registered model or fine-tuned model
188-
if (
189-
service_model.freeform_tags.get(Tags.BASE_MODEL_CUSTOM, None) is not None
190-
or service_model.freeform_tags.get(Tags.AQUA_FINE_TUNED_MODEL_TAG)
191-
is not None
192-
):
188+
# Skip model copying if it is registered model
189+
if service_model.freeform_tags.get(Tags.BASE_MODEL_CUSTOM, None) is not None:
193190
logger.info(
194191
f"Aqua Model {model_id} already exists in the user's compartment."
195192
"Skipped copying."
@@ -716,7 +713,8 @@ def _extract_model_task(
716713
model.model_task = task_tag
717714
else:
718715
raise AquaValueError(
719-
f"{task_tag} is not supported. Valid model_task inputs are: {MultiModelSupportedTaskType.values()}."
716+
f"Invalid or missing {task_tag} tag for selected model {source_model.display_name}. "
717+
f"Currently only `{MultiModelSupportedTaskType.values()}` models are supported for multi model deployment."
720718
)
721719

722720
def _fetch_metric_from_metadata(

0 commit comments

Comments
 (0)