feat: initial code for fal.ai integration
This commit is contained in:
@@ -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.
|
||||
|
Reference in New Issue
Block a user