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

Merge pull request #23 from ace-step/fix_extend_bug

Browse Source 
fix extend: torch.cate -> torch.cat
This commit is contained in:
Gong Junmin
2025-05-07 13:33:20 +08:00
committed by GitHub
parent ce02ddca99 cdc148f49c
commit 2b8d33a915
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pipeline_ace_step.py
+2 -2
View File
@@ -912,9 +912,9 @@ class ACEStepPipeline:
if is_extend: if is_extend:
if to_right_pad_gt_latents is not None: if to_right_pad_gt_latents is not None:
target_latents = torch.cate([target_latents, to_right_pad_gt_latents], dim=-1) target_latents = torch.cat([target_latents, to_right_pad_gt_latents], dim=-1)
if to_left_pad_gt_latents is not None: if to_left_pad_gt_latents is not None:
target_latents = torch.cate([to_right_pad_gt_latents, target_latents], dim=0) target_latents = torch.cat([to_right_pad_gt_latents, target_latents], dim=0)
return target_latents return target_latents
def latents2audio(self, latents, target_wav_duration_second=30, sample_rate=48000, save_path=None, format="flac"): def latents2audio(self, latents, target_wav_duration_second=30, sample_rate=48000, save_path=None, format="flac"):