feat: initial code for fal.ai integration

This commit is contained in:
2024-07-25 10:36:24 +01:00
parent 9869c3524e
commit 0e744af73f
4 changed files with 48 additions and 9 deletions

View File

@@ -2,6 +2,8 @@ import torchaudio
from audiocraft.models.musicgen import MusicGen
from audiocraft.data.audio import audio_write
from prompts import PROMPTS
MODEL_NAME = "facebook/musicgen-large"
MUSIC_DURATION_SECONDS = 60
@@ -17,7 +19,7 @@ descriptions = [
def generate(offset=0):
wav = model.generate(descriptions)
wav = model.generate(PROMPTS)
for idx, one_wav in enumerate(wav):
# Will save under {idx}.wav, with loudness normalization at -14 db LUFS.