@huggingface/tasks 0.3.3 → 0.3.4

package/dist/index.cjs

@@ -1295,7 +1295,7 @@ var PIPELINE_DATA = {
     color: "green"
   },
   "image-text-to-text": {
-    name: "Image + Text to Text (VLLMs)",
+    name: "Image-Text-to-Text",
     modality: "multimodal",
     color: "red",
     hideInDatasets: true
@@ -4441,6 +4441,41 @@ var pythae = (model) => [
 
 model = AutoModel.load_from_hf_hub("${model.id}")`
 ];
+var musicgen = (model) => [
+  `from audiocraft.models import MusicGen
+
+model = MusicGen.get_pretrained("${model.id}")
+
+descriptions = ['happy rock', 'energetic EDM', 'sad jazz']
+wav = model.generate(descriptions)  # generates 3 samples.`
+];
+var magnet = (model) => [
+  `from audiocraft.models import MAGNeT
+	
+model = MAGNeT.get_pretrained("${model.id}")
+
+descriptions = ['disco beat', 'energetic EDM', 'funky groove']
+wav = model.generate(descriptions)  # generates 3 samples.`
+];
+var audiogen = (model) => [
+  `from audiocraft.models import AudioGen
+	
+model = AudioGen.get_pretrained("${model.id}")
+model.set_generation_params(duration=5)  # generate 5 seconds.
+descriptions = ['dog barking', 'sirene of an emergency vehicle', 'footsteps in a corridor']
+wav = model.generate(descriptions)  # generates 3 samples.`
+];
+var audiocraft = (model) => {
+  if (model.tags?.includes("musicgen")) {
+    return musicgen(model);
+  } else if (model.tags?.includes("audiogen")) {
+    return audiogen(model);
+  } else if (model.tags?.includes("magnet")) {
+    return magnet(model);
+  } else {
+    return [`# Type of model unknown.`];
+  }
+};
 
 // src/model-libraries.ts
 var MODEL_LIBRARIES_UI_ELEMENTS = {
@@ -4474,6 +4509,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
       term: { path: "pytorch_model.bin" }
     }
   },
+  audiocraft: {
+    prettyLabel: "Audiocraft",
+    repoName: "audiocraft",
+    repoUrl: "https://github.com/facebookresearch/audiocraft",
+    snippets: audiocraft,
+    filter: false
+  },
   bertopic: {
     prettyLabel: "BERTopic",
     repoName: "BERTopic",