diff --git a/src/controller/artifact/processor/hf/hf.go b/src/controller/artifact/processor/hf/hf.go index 1356e0774cf..99a17e525a6 100644 --- a/src/controller/artifact/processor/hf/hf.go +++ b/src/controller/artifact/processor/hf/hf.go @@ -26,7 +26,7 @@ import ( // const definitions const ( ArtifactTypeHF = "AIArtifact" - MediaType = "application/vnd.goharbor.huggingface.v1" + MediaType = "application/vnd.goharbor.aiartifact.v1+json" ) func init() { diff --git a/src/controller/replication/transfer/huggingface/transfer.go b/src/controller/replication/transfer/huggingface/transfer.go index 36172d068c6..c1978a4ace3 100644 --- a/src/controller/replication/transfer/huggingface/transfer.go +++ b/src/controller/replication/transfer/huggingface/transfer.go @@ -245,9 +245,9 @@ func (t *transfer) composeOCI(ctx context.Context, fs *file.Store, files []strin artifactType := hf.MediaType orasOpts := oras.PackManifestOptions{ Layers: fileDescriptors, - //ConfigDescriptor: &v1.Descriptor{ - // MediaType: hf.MediaType, - //}, + ConfigDescriptor: &v1.Descriptor{ + MediaType: hf.MediaType, + }, ManifestAnnotations: map[string]string{ "type": "hugging-face-object", "author": "Stability AI",