theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-04-02 15:48:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

add acestep models

Browse Source
This commit is contained in:
Artiprocher
2026-04-02 10:58:45 +08:00
parent 166e6d2d38
commit 59f512b574
8 changed files with 3169 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
examples/ace_step/model_inference/ace_step.py Normal file
View File

@@ -0,0 +1,14 @@
from diffsynth.pipelines.ace_step_audio import AceStepAudioPipeline, ModelConfig
import torch
pipe = AceStepAudioPipeline.from_pretrained(
torch_dtype=torch.bfloat16,
device="cuda",
model_configs=[
ModelConfig(model_id="ACE-Step/Ace-Step1.5", origin_file_pattern="Qwen3-Embedding-0.6B/model.safetensors"),
ModelConfig(model_id="ACE-Step/Ace-Step1.5", origin_file_pattern="acestep-v15-turbo/model.safetensors"),
ModelConfig(model_id="ACE-Step/Ace-Step1.5", origin_file_pattern="vae/diffusion_pytorch_model.safetensors"),
],
tokenizer_config=ModelConfig(model_id="ACE-Step/Ace-Step1.5", origin_file_pattern="Qwen3-Embedding-0.6B"),
)