fix

wy65701436 · Jul 3, 2024 · a8e7d2c · a8e7d2c
1 parent d4db0fd
commit a8e7d2c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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",