add stable audio small pingpong sampler support

This commit is contained in:
chuxij
2025-05-15 06:35:17 +00:00
parent b313d05299
commit 05c58339f0
3 changed files with 353 additions and 4 deletions
+4 -4
View File
@@ -97,8 +97,7 @@ def create_text2music_ui(
# Get base output directory from environment variable, defaulting to CWD-relative 'outputs'.
# This default (./outputs) is suitable for non-Docker local development.
# For Docker, the ACE_OUTPUT_DIR environment variable should be set (e.g., to /app/outputs).
_output_base_dir = os.environ.get("ACE_OUTPUT_DIR", "./outputs")
output_file_dir = os.path.join(_output_base_dir, "text2music")
output_file_dir = os.environ.get("ACE_OUTPUT_DIR", "./outputs")
if not os.path.isdir(output_file_dir):
os.makedirs(output_file_dir, exist_ok=True)
json_files = [f for f in os.listdir(output_file_dir) if f.endswith('.json')]
@@ -235,7 +234,7 @@ def create_text2music_ui(
with gr.Accordion("Advanced Settings", open=False):
scheduler_type = gr.Radio(
["euler", "heun"],
["euler", "heun", "pingpong"],
value="euler",
label="Scheduler Type",
elem_id="scheduler_type",
@@ -911,7 +910,8 @@ def create_text2music_ui(
)
def load_data(json_file):
json_file = os.path.join(output_file_dir, json_file)
if isinstance(output_file_dir, str):
json_file = os.path.join(output_file_dir, json_file)
json_data = load_data_func(json_file)
return json2output(json_data)