Skip to content

Commit c426be3

Browse files
authored
Resolve ACE Step dependencies (#1424)
* remove vector_quantize_pytorch dep * remove einx
1 parent 534e2e4 commit c426be3

3 files changed

Lines changed: 572 additions & 5 deletions

File tree

diffsynth/configs/vram_management_module_maps.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -327,7 +327,7 @@
327327
"diffsynth.models.ace_step_tokenizer.AceStepTokenizer": {
328328
"torch.nn.Linear": "diffsynth.core.vram.layers.AutoWrappedLinear",
329329
"torch.nn.Embedding": "diffsynth.core.vram.layers.AutoWrappedModule",
330-
"vector_quantize_pytorch.ResidualFSQ": "diffsynth.core.vram.layers.AutoWrappedModule",
330+
"diffsynth.models.ace_step_residual_fsq.ResidualFSQ": "diffsynth.core.vram.layers.AutoWrappedModule",
331331
"transformers.models.qwen3.modeling_qwen3.Qwen3RMSNorm": "diffsynth.core.vram.layers.AutoWrappedModule",
332332
"transformers.models.qwen3.modeling_qwen3.Qwen3MLP": "diffsynth.core.vram.layers.AutoWrappedModule",
333333
"transformers.models.qwen3.modeling_qwen3.Qwen3RotaryEmbedding": "diffsynth.core.vram.layers.AutoWrappedModule",

0 commit comments

Comments
 (0)