INNER CODE UNIT · Python

music_audiogen

Woolverine94/biniou · ressources/audiogen.py:44

def music_audiogen(
    prompt_audiogen, 
    model_audiogen, 
    duration_audiogen, 
    num_batch_audiogen, 
    temperature_audiogen, 
    top_k_audiogen, 
    top_p_audiogen, 
    use_sampling_audiogen, 
    cfg_coef_audiogen, 
    progress_audiogen=gr.Progress(track_tqdm=True)
    ):

    print(">>>[AudioGen 🔊 ]: starting module")

    pipe_audiogen = AudioGen.get_pretrained(model_audiogen, device=device_audiogen)
    pipe_audiogen.set_generation_params(
        duration=duration_audiogen, 

View source record →