Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 5 additions & 2 deletions ads/aqua/model/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -660,7 +660,7 @@ def edit_registered_model(
raise AquaRuntimeError("Only registered unverified models can be edited.")

def convert_fine_tune(
self, model_id: str, delete_model: Optional[bool] = False
self, model_id: str, delete_model: Optional[bool] = True
) -> DataScienceModel:
"""Converts legacy fine tuned model to fine tuned model v2.
1. 'fine_tune_model_version' tag will be added as 'v2' to new fine tuned model.
Expand All @@ -671,7 +671,7 @@ def convert_fine_tune(
model_id: str
The legacy fine tuned model OCID.
delete_model: bool
Flag whether to delete the legacy model or not. Defaults to False.
Flag whether to delete the legacy model or not. Defaults to True.

Returns
-------
Expand Down Expand Up @@ -738,6 +738,9 @@ def convert_fine_tune(
)

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

return fine_tune_model_v2
Expand Down