kennethnym/infinifi
1
0
Fork 0
Code Issues 2 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

feat: gen more audio files and increase duration

Browse Source
This commit is contained in:
Kenneth
2024-07-20 16:28:16 +01:00
parent ec71c0b235
commit 91f1db90a3
1 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
generate.py
View File

@@ -1,19 +1,32 @@
import torchaudio import torchaudio
import time
from audiocraft.models.musicgen import MusicGen from audiocraft.models.musicgen import MusicGen
from audiocraft.data.audio import audio_write from audiocraft.data.audio import audio_write
MODEL_NAME = "facebook/musicgen-large" MODEL_NAME = "facebook/musicgen-large"
MUSIC_DURATION_SECONDS = 60
print(f"getting {MODEL_NAME}...") print(f"getting {MODEL_NAME}...")
model = MusicGen.get_pretrained(MODEL_NAME) model = MusicGen.get_pretrained(MODEL_NAME)
descriptions = ["gentle lo-fi beats"] model.set_generation_params(duration=MUSIC_DURATION_SECONDS)
descriptions = [
"gentle, calming lo-fi beats that helps with studying and focusing",
"calm, piano lo-fi beats to help with studying and focusing",
"gentle lo-fi hip-hop to relax to",
"gentle, quiet synthwave lo-fi beats",
"morning lo-fi beats"
]
print("model obtained. generating wav files...") print("model obtained. generating wav files...")
a = time.time()
wav = model.generate(descriptions) wav = model.generate(descriptions)
print(f"{len(wav)} generated.") b = time.time()
print(f"{len(wav)} generated. took {b - a} seconds.")
for idx, one_wav in enumerate(wav): for idx, one_wav in enumerate(wav):
# Will save under {idx}.wav, with loudness normalization at -14 db LUFS. # Will save under {idx}.wav, with loudness normalization at -14 db LUFS.