File tree Expand file tree Collapse file tree 1 file changed +9
-2
lines changed Expand file tree Collapse file tree 1 file changed +9
-2
lines changed Original file line number Diff line number Diff 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 {}))
You can’t perform that action at this time.
0 commit comments