Spaces:
Running
on
A100
Running
on
A100
Update app.py
Browse files
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"
|
116 |
-
"--lyrics_txt", f"
|
117 |
"--run_n_segments", str(num_segments),
|
118 |
"--stage2_batch_size", "4",
|
119 |
-
"--output_dir", f"
|
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"
|