add stable audio small pingpong sampler support
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user