theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-04-24 06:46:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

acestep t2m

Browse Source
This commit is contained in:
mi804
2026-04-21 13:16:15 +08:00
parent a604d76339
commit 9d09e0431c
9 changed files with 300 additions and 377 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
diffsynth/models/ace_step_dit.py
View File

@@ -165,7 +165,7 @@ class TimestepEmbedding(nn.Module):
self,
in_channels: int,
time_embed_dim: int,
scale: float = 1000,
scale: float = 1,
):
super().__init__()
@@ -711,7 +711,7 @@ class AceStepDiTModel(nn.Module):
encoder_hidden_states: torch.Tensor,
encoder_attention_mask: torch.Tensor,
context_latents: torch.Tensor,
use_cache: Optional[bool] = None,
use_cache: Optional[bool] = False,
past_key_values: Optional[EncoderDecoderCache] = None,
cache_position: Optional[torch.LongTensor] = None,
position_ids: Optional[torch.LongTensor] = None,