fffiloni commited on
Commit
c193789
·
verified ·
1 Parent(s): db2be5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -112,11 +112,11 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
112
  "python", "infer.py",
113
  "--stage1_model", "m-a-p/YuE-s1-7B-anneal-en-cot",
114
  "--stage2_model", "m-a-p/YuE-s2-1B-general",
115
- "--genre_txt", f"'{genre_txt_path}'",
116
- "--lyrics_txt", f"'{lyrics_txt_path}'",
117
  "--run_n_segments", str(num_segments),
118
  "--stage2_batch_size", "4",
119
- "--output_dir", f"'{output_dir}'",
120
  "--cuda_idx", "0",
121
  "--max_new_tokens", str(max_new_tokens),
122
  "--disable_offload_model"
 
112
  "python", "infer.py",
113
  "--stage1_model", "m-a-p/YuE-s1-7B-anneal-en-cot",
114
  "--stage2_model", "m-a-p/YuE-s2-1B-general",
115
+ "--genre_txt", f"{genre_txt_path}",
116
+ "--lyrics_txt", f"{lyrics_txt_path}",
117
  "--run_n_segments", str(num_segments),
118
  "--stage2_batch_size", "4",
119
+ "--output_dir", f"{output_dir}",
120
  "--cuda_idx", "0",
121
  "--max_new_tokens", str(max_new_tokens),
122
  "--disable_offload_model"