File tree 1 file changed +5
-7
lines changed
1 file changed +5
-7
lines changed Original file line number Diff line number Diff line change 81
81
ImportModelDetails ,
82
82
ModelValidationResult ,
83
83
)
84
+ from ads .aqua .model .enums import MultiModelSupportedTaskType
84
85
from ads .common .auth import default_signer
85
86
from ads .common .oci_resource import SEARCH_TYPE , OCIResource
86
87
from ads .common .utils import (
@@ -184,12 +185,8 @@ def create(
184
185
target_project = project_id or PROJECT_OCID
185
186
target_compartment = compartment_id or COMPARTMENT_OCID
186
187
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 :
193
190
logger .info (
194
191
f"Aqua Model { model_id } already exists in the user's compartment."
195
192
"Skipped copying."
@@ -716,7 +713,8 @@ def _extract_model_task(
716
713
model .model_task = task_tag
717
714
else :
718
715
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."
720
718
)
721
719
722
720
def _fetch_metric_from_metadata (
You can’t perform that action at this time.
0 commit comments