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 committed 2025-04-28 15:21:19 +00:00
1 parent 892e1b56e7
commit 09560dd878
7 files changed
+47 -49
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

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