cocktailpeanut commited on
Commit
4279593
Β·
1 Parent(s): b47e0bf
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -48,9 +48,12 @@ def infer_music(lrc, ref_audio_path, steps, file_type, cfg_strength, odeint_meth
48
  lrc_prompt, start_time = get_lrc_token(max_frames, lrc, tokenizer, device)
49
  # style_prompt = get_style_prompt(muq, ref_audio_path, prompt)
50
 
 
51
  if prompt is not None:
 
52
  style_prompt = get_text_style_prompt(muq, prompt)
53
  else:
 
54
  style_prompt, vocal_flag = get_audio_style_prompt(muq, ref_audio_path)
55
 
56
  negative_style_prompt = get_negative_style_prompt(device)
 
48
  lrc_prompt, start_time = get_lrc_token(max_frames, lrc, tokenizer, device)
49
  # style_prompt = get_style_prompt(muq, ref_audio_path, prompt)
50
 
51
+ print(f"infer_music: lrc={lrc}, ref_audio_path={ref_audio_path}, steps={steps}, file_type={file_type}, cfg_strength={cfg_strength}, odeint_method={odeint_method}, duration={duration}, prompt={prompt}")
52
  if prompt is not None:
53
+ print("text style prompt")
54
  style_prompt = get_text_style_prompt(muq, prompt)
55
  else:
56
+ print("audio style prompt")
57
  style_prompt, vocal_flag = get_audio_style_prompt(muq, ref_audio_path)
58
 
59
  negative_style_prompt = get_negative_style_prompt(device)