mrleo1nid/ACE-Step
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rename to ace step

Browse Source
This commit is contained in:
chuxij
2025-04-28 15:21:19 +00:00
parent 892e1b56e7
commit 09560dd878
7 changed files with 47 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/fusic_transformer.py → models/ace_step_transformer.py
+1 -1
View File
@@ -153,7 +153,7 @@ class Transformer2DModelOutput(BaseOutput):
proj_losses: Optional[Tuple[Tuple[str, torch.Tensor]]] = None
class FusicTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin, FromOriginalModelMixin):
class ACEStepTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin, FromOriginalModelMixin):
_supports_gradient_checkpointing = True
@register_to_config