wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-25 19:13:31 +08:00
Code Issues Actions 27 Packages Projects Releases Wiki Activity

Fix lint: whitespace and unused vars

Browse Source
This commit is contained in:
qqingzheng 2026-03-10 15:36:37 +08:00
parent 457e7000e2
commit b476d5e62b
2 changed files with 7 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
comfy/ldm/helios/model.py
View File

@ -711,19 +711,9 @@ class HeliosModel(torch.nn.Module):
) )
f_long = self._rope_downsample_3d(f_long, (long_t, hs, ws), (4, 4, 4)) f_long = self._rope_downsample_3d(f_long, (long_t, hs, ws), (4, 4, 4))
hidden_states = torch.cat([x_long, hidden_states], dim=1) hidden_states = torch.cat([x_long, hidden_states], dim=1)
freqs = torch.cat([f_long, freqs], dim=1) freqs = torch.cat([f_long, freqs], dim=1)
history_context_length = hidden_states.shape[1] - original_context_length history_context_length = hidden_states.shape[1] - original_context_length
mismatch = hidden_states.shape[1] != freqs.shape[1]
summary_key = (
int(post_t),
int(post_h),
int(post_w),
int(original_context_length),
int(hidden_states.shape[1]),
int(freqs.shape[1]),
int(history_context_length),
)
if timestep.ndim == 0: if timestep.ndim == 0:
timestep = timestep.unsqueeze(0) timestep = timestep.unsqueeze(0)

1
comfy_extras/nodes_helios.py
View File

@ -412,7 +412,6 @@ def _helios_dmd_sample(
for i in range(len(sigmas) - 1): for i in range(len(sigmas) - 1):
sigma = sigmas[i] sigma = sigmas[i]
sigma_next = sigmas[i + 1]
timestep = all_timesteps[i] if i < len(all_timesteps) else i timestep = all_timesteps[i] if i < len(all_timesteps) else i
denoised = model(x, sigma * s_in, **extra_args) denoised = model(x, sigma * s_in, **extra_args)