Skip to content

Commit 45bb675

Browse files
Updating skip model creation check
1 parent f6eb555 commit 45bb675

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

ads/aqua/model/model.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,6 @@
8080
ImportModelDetails,
8181
ModelValidationResult,
8282
)
83-
from ads.aqua.model.enums import MultiModelSupportedTaskType
8483
from ads.common.auth import default_signer
8584
from ads.common.oci_resource import SEARCH_TYPE, OCIResource
8685
from ads.common.utils import (
@@ -184,8 +183,12 @@ def create(
184183
target_project = project_id or PROJECT_OCID
185184
target_compartment = compartment_id or COMPARTMENT_OCID
186185

187-
# Skip model copying if it is registered model
188-
if service_model.freeform_tags.get(Tags.BASE_MODEL_CUSTOM, None) is not None:
186+
# Skip model copying if it is registered model or fine-tuned model
187+
if (
188+
service_model.freeform_tags.get(Tags.BASE_MODEL_CUSTOM, None) is not None
189+
or service_model.freeform_tags.get(Tags.AQUA_FINE_TUNED_MODEL_TAG)
190+
is not None
191+
):
189192
logger.info(
190193
f"Aqua Model {model_id} already exists in the user's compartment."
191194
"Skipped copying."

0 commit comments

Comments
 (0)