wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-20 21:38:30 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Merge 0975da3d75 into 8edea4a65d

Browse Source
This commit is contained in:
Yuchen Fan 2026-07-18 00:29:32 +08:00 committed by GitHub
commit 20a110fa75
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/model_base.py
View File

@ -801,7 +801,8 @@ class StableAudio1(BaseModel):
device = kwargs["device"] device = kwargs["device"]
seconds_start = kwargs.get("seconds_start", 0) seconds_start = kwargs.get("seconds_start", 0)
seconds_total = kwargs.get("seconds_total", int(noise.shape[-1] / 21.53)) ratio = self.latent_format.temporal_downscale_ratio
seconds_total = kwargs.get("seconds_total", round(noise.shape[-1] * ratio / 44100))
seconds_start_embed = self.seconds_start_embedder([seconds_start])[0].to(device) seconds_start_embed = self.seconds_start_embedder([seconds_start])[0].to(device)
seconds_total_embed = self.seconds_total_embedder([seconds_total])[0].to(device) seconds_total_embed = self.seconds_total_embedder([seconds_total])[0].to(device)
@ -868,7 +869,8 @@ class StableAudio3(BaseModel):
noise = kwargs.get("noise", None) noise = kwargs.get("noise", None)
device = kwargs["device"] device = kwargs["device"]
seconds_total = kwargs.get("seconds_total", int(noise.shape[-1] / 10.7666)) ratio = self.latent_format.temporal_downscale_ratio
seconds_total = kwargs.get("seconds_total", round(noise.shape[-1] * ratio / 44100))
seconds_total_embed = self.seconds_total_embedder([seconds_total])[0].to(device) seconds_total_embed = self.seconds_total_embedder([seconds_total])[0].to(device)
global_embed = seconds_total_embed.reshape((1, -1)) global_embed = seconds_total_embed.reshape((1, -1))