INNER CODE UNIT · Python
check_audiogen
Woolverine94/biniou · ressources/audiogen.py:30
def check_audiogen(generated_tokens, total_tokens) :
global stop_audiogen
if stop_audiogen == False :
return
elif stop_audiogen == True :
print(">>>[AudioGen 🔊 ]: generation canceled by user")
stop_audiogen = False
try:
del ressources.audiogen.pipe_audiogen
except NameError as e:
raise Exception("Interrupting ...")
return
@metrics_decoration
def music_audiogen(
prompt_audiogen,
model_audiogen,
duration_audiogen,