Skip to content

Commit 93efdca

Browse files
committed
Updated convert fine tune.
1 parent 6aab419 commit 93efdca

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

ads/aqua/model/model.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -660,7 +660,7 @@ def edit_registered_model(
660660
raise AquaRuntimeError("Only registered unverified models can be edited.")
661661

662662
def convert_fine_tune(
663-
self, model_id: str, delete_model: Optional[bool] = False
663+
self, model_id: str, delete_model: Optional[bool] = True
664664
) -> DataScienceModel:
665665
"""Converts legacy fine tuned model to fine tuned model v2.
666666
1. 'fine_tune_model_version' tag will be added as 'v2' to new fine tuned model.
@@ -671,7 +671,7 @@ def convert_fine_tune(
671671
model_id: str
672672
The legacy fine tuned model OCID.
673673
delete_model: bool
674-
Flag whether to delete the legacy model or not. Defaults to False.
674+
Flag whether to delete the legacy model or not. Defaults to True.
675675
676676
Returns
677677
-------
@@ -738,6 +738,9 @@ def convert_fine_tune(
738738
)
739739

740740
if delete_model:
741+
logger.info(
742+
f"Deleting legacy model {model_id}. To keep both models next time, set 'delete_model' as 'False'."
743+
)
741744
legacy_fine_tuned_model.delete()
742745

743746
return fine_tune_model_v2

0 commit comments

Comments
 (0)