modified: app.py

app.py

@@ -128,7 +128,7 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=
         "--genre_txt", f"{genre_txt_path}",
         "--lyrics_txt", f"{lyrics_txt_path}",
         "--run_n_segments", f"{num_segments}",
-        "--stage2_batch_size", "4",
+        # "--stage2_batch_size", "4",
         "--output_dir", f"{output_dir}",
         "--cuda_idx", "0",
         "--max_new_tokens", f"{max_new_tokens}",