ChuxiJ commited on
Commit
9d632db
·
1 Parent(s): d32ba2d

feat: add XL (4B DiT) turbo as default model + acemusic banner

Browse files

- Default model 1: acestep-v15-xl-turbo (was acestep-v15-turbo)
- Default model 2: acestep-v15-turbo (was acestep-v15-turbo-shift3)
- Add XL model entries to MODEL_REPO_MAPPING for auto-download
- Enhanced acemusic.ai banner with gradient background
- Update README frontmatter with XL model

README.md CHANGED
@@ -9,6 +9,7 @@ python_version: 3.11
9
  pinned: false
10
  models:
11
  - ACE-Step/Ace-Step1.5
 
12
  license: mit
13
  app_file: app.py
14
  short_description: Music Generation Foundation Model v1.5
 
9
  pinned: false
10
  models:
11
  - ACE-Step/Ace-Step1.5
12
+ - ACE-Step/acestep-v15-xl-turbo
13
  license: mit
14
  app_file: app.py
15
  short_description: Music Generation Foundation Model v1.5
acestep/gradio_ui/interfaces/__init__.py CHANGED
@@ -64,7 +64,14 @@ def create_gradio_interface(dit_handler, llm_handler, dataset_handler, init_para
64
  gr.HTML(f"""
65
  <div class="main-header">
66
  <h1>{t("app.title")}</h1>
67
- <p>{t("app.subtitle")}, try more on <a href="https://acemusic.ai" target="_blank">acemusic.ai</a></p>
 
 
 
 
 
 
 
68
  <p style="margin-top: 0.5rem;">
69
  <a href="https://ace-step.github.io/ace-step-v1.5.github.io/" target="_blank">Project</a> |
70
  <a href="https://huggingface.co/collections/ACE-Step/ace-step-15" target="_blank">Hugging Face</a> |
 
64
  gr.HTML(f"""
65
  <div class="main-header">
66
  <h1>{t("app.title")}</h1>
67
+ <p>{t("app.subtitle")}</p>
68
+ <div style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); padding: 10px 20px; border-radius: 8px; text-align: center; margin: 8px auto; max-width: 600px;">
69
+ <span style="color: white; font-size: 15px;">
70
+ 🚀 Want faster &amp; more stable experience? Try
71
+ <a href="https://acemusic.ai" target="_blank" style="color: #ffd700; font-weight: bold; text-decoration: underline;">acemusic.ai</a>
72
+ — 100% free!
73
+ </span>
74
+ </div>
75
  <p style="margin-top: 0.5rem;">
76
  <a href="https://ace-step.github.io/ace-step-v1.5.github.io/" target="_blank">Project</a> |
77
  <a href="https://huggingface.co/collections/ACE-Step/ace-step-15" target="_blank">Hugging Face</a> |
acestep/handler.py CHANGED
@@ -135,6 +135,10 @@ class AceStepHandler:
135
  "acestep-v15-base": "ACE-Step/acestep-v15-base",
136
  "acestep-v15-sft": "ACE-Step/acestep-v15-sft",
137
  "acestep-v15-turbo-shift3": "ACE-Step/acestep-v15-turbo-shift3",
 
 
 
 
138
  }
139
 
140
  # Default fallback repository for unknown models
 
135
  "acestep-v15-base": "ACE-Step/acestep-v15-base",
136
  "acestep-v15-sft": "ACE-Step/acestep-v15-sft",
137
  "acestep-v15-turbo-shift3": "ACE-Step/acestep-v15-turbo-shift3",
138
+ # XL (4B) DiT models
139
+ "acestep-v15-xl-turbo": "ACE-Step/acestep-v15-xl-turbo",
140
+ "acestep-v15-xl-base": "ACE-Step/acestep-v15-xl-base",
141
+ "acestep-v15-xl-sft": "ACE-Step/acestep-v15-xl-sft",
142
  }
143
 
144
  # Default fallback repository for unknown models
app.py CHANGED
@@ -166,10 +166,10 @@ def main():
166
  # Service mode configuration from environment variables
167
  config_path = os.environ.get(
168
  "SERVICE_MODE_DIT_MODEL",
169
- "acestep-v15-turbo"
170
  )
171
  # Second DiT model - default to turbo-shift3 for two-model setup
172
- config_path_2 = os.environ.get("SERVICE_MODE_DIT_MODEL_2", "acestep-v15-turbo-shift3").strip()
173
 
174
  lm_model_path = os.environ.get(
175
  "SERVICE_MODE_LM_MODEL",
 
166
  # Service mode configuration from environment variables
167
  config_path = os.environ.get(
168
  "SERVICE_MODE_DIT_MODEL",
169
+ "acestep-v15-xl-turbo"
170
  )
171
  # Second DiT model - default to turbo-shift3 for two-model setup
172
+ config_path_2 = os.environ.get("SERVICE_MODE_DIT_MODEL_2", "acestep-v15-turbo").strip()
173
 
174
  lm_model_path = os.environ.get(
175
  "SERVICE_MODE_LM_MODEL",