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

Merge pull request #37 from fakerybakery/fix-torch-cat

Browse Source 
torch.cate -> torch.cat
This commit is contained in:
Sayo
2025-05-07 23:42:34 +08:00
committed by GitHub
parent be5c795257 fc2cfa2d93
commit 31d81a5040
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
acestep/pipeline_ace_step.py
+2 -2
View File
@@ -1313,11 +1313,11 @@ 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 = torch.cat(
[target_latents, to_right_pad_gt_latents], dim=-1 [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( target_latents = torch.cat(
[to_right_pad_gt_latents, target_latents], dim=0 [to_right_pad_gt_latents, target_latents], dim=0
) )
return target_latents return target_latents