commit 0ff74a85ef43cdd852c17421f3feaea7e265bad7
Author: Kenneth <kennethnym@outlook.com>
Date:   Sat Jul 20 12:24:31 2024 +0100

    initial commit

diff --git a/generate.py b/generate.py
new file mode 100644
index 0000000..74818c1
--- /dev/null
+++ b/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,
+    )