Skip to content

Commit 6e489b7

Browse files
authored
Update model.py
1 parent 411a47c commit 6e489b7

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

ads/aqua/model/model.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -660,7 +660,10 @@ 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] = True
663+
self,
664+
model_id: str,
665+
delete_model: Optional[bool] = True,
666+
model_display_name: Optional[str] = None,
664667
) -> DataScienceModel:
665668
"""Converts legacy fine tuned model to fine tuned model v2.
666669
1. 'fine_tune_model_version' tag will be added as 'v2' to new fine tuned model.
@@ -672,6 +675,8 @@ def convert_fine_tune(
672675
The legacy fine tuned model OCID.
673676
delete_model: bool
674677
Flag whether to delete the legacy model or not. Defaults to True.
678+
model_display_name: str
679+
The name of fine tuned model v2 converted. Legacy model's name will be used if not provided. Defaults to None.
675680
676681
Returns
677682
-------
@@ -722,7 +727,9 @@ def convert_fine_tune(
722727
.with_model_file_description(
723728
json_dict=legacy_fine_tuned_model.model_file_description
724729
)
725-
.with_display_name(legacy_fine_tuned_model.display_name)
730+
.with_display_name(
731+
model_display_name or legacy_fine_tuned_model.display_name
732+
)
726733
.with_description(legacy_fine_tuned_model.description)
727734
.with_freeform_tags(**fine_tune_model_v2_tags)
728735
.with_defined_tags(**(legacy_fine_tuned_model.defined_tags or {}))

0 commit comments

Comments
 (0)