@huggingface/tasks 0.3.3 → 0.4.0

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
@@ -4222,7 +4222,7 @@ model = joblib.load(
 };
 var sklearn = (model) => {
   if (model.tags?.includes("skops")) {
-    const skopsmodelFile = model.config?.sklearn?.filename;
+    const skopsmodelFile = model.config?.sklearn?.model?.file;
     const skopssaveFormat = model.config?.sklearn?.model_format;
     if (!skopsmodelFile) {
       return [`# \u26A0\uFE0F Model filename not specified in config.json`];
@@ -4292,7 +4292,7 @@ var speechBrainMethod = (speechbrainInterface) => {
   }
 };
 var speechbrain = (model) => {
-  const speechbrainInterface = model.config?.speechbrain?.interface;
+  const speechbrainInterface = model.config?.speechbrain?.speechbrain_interface;
   if (speechbrainInterface === void 0) {
     return [`# interface not specified in config.json`];
   }
@@ -4370,7 +4370,7 @@ var peftTask = (peftTaskType) => {
   }
 };
 var peft = (model) => {
-  const { base_model_name: peftBaseModel, task_type: peftTaskType } = model.config?.peft ?? {};
+  const { base_model_name_or_path: peftBaseModel, task_type: peftTaskType } = model.config?.peft ?? {};
   const pefttask = peftTask(peftTaskType);
   if (!pefttask) {
     return [`Task type is invalid.`];
@@ -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",