initial commit

generate.py

@@ -0,0 +1,26 @@
+import torchaudio
+from audiocraft.models.magnet import MAGNeT
+from audiocraft.data.audio import audio_write
+
+MODEL_NAME = "facebook/magnet-small-10secs"
+
+print(f"getting {MODEL_NAME}...")
+
+model = MAGNeT.get_pretrained(MODEL_NAME)
+descriptions = ["gentle lo-fi beats"]
+
+print("model obtained. generating wav files...")
+
+wav = model.generate(descriptions)
+
+print(f"{len(wav)} generated.")
+
+for idx, one_wav in enumerate(wav):
+    # Will save under {idx}.wav, with loudness normalization at -14 db LUFS.
+    audio_write(
+        f"{idx}",
+        one_wav.cpu(),
+        model.sample_rate,
+        strategy="loudness",
+        loudness_compressor=True,
+    )