Skip to content

Commit f6eb555

Browse files
Updating finetuning flow for DRCC
1 parent 9b0a249 commit f6eb555

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

ads/aqua/finetuning/finetuning.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@
5858
from ads.jobs.builders.infrastructure.dsc_job import DataScienceJob
5959
from ads.jobs.builders.runtimes.base import Runtime
6060
from ads.jobs.builders.runtimes.container_runtime import ContainerRuntime
61+
from ads.model.common.utils import MetadataArtifactPathType
6162
from ads.model.model_metadata import (
6263
MetadataTaxonomyKeys,
6364
ModelCustomMetadata,
@@ -315,6 +316,23 @@ def create(
315316
model_by_reference=True,
316317
defined_tags=create_fine_tuning_details.defined_tags,
317318
)
319+
defined_metadata_dict = {}
320+
defined_metadata_list_source = source.defined_metadata_list._to_oci_metadata()
321+
for defined_metadata in defined_metadata_list_source:
322+
if (
323+
defined_metadata.has_artifact
324+
and defined_metadata.key
325+
!= AquaModelMetadataKeys.FINE_TUNING_CONFIGURATION
326+
):
327+
content = self.ds_client.get_model_defined_metadatum_artifact_content(
328+
source.id, defined_metadata.key
329+
).data.content
330+
defined_metadata_dict[defined_metadata.key] = content
331+
332+
for key, value in defined_metadata_dict.items():
333+
ft_model.create_defined_metadata_artifact(
334+
key, value, MetadataArtifactPathType.CONTENT
335+
)
318336

319337
ft_job_freeform_tags = {
320338
Tags.AQUA_TAG: UNKNOWN,

0 commit comments

Comments
 (0)