wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-10 05:22:34 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Fix ruff.

Browse Source
This commit is contained in:
comfyanonymous 2025-12-02 21:29:26 -05:00 committed by GitHub
parent fd0c77b606
commit 55e8737dc5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/lumina/controlnet.py
View File

@ -1,6 +1,5 @@
import torch import torch
from torch import nn from torch import nn
import math
from .model import JointTransformerBlock from .model import JointTransformerBlock
@ -102,7 +101,6 @@ class ZImage_Control(torch.nn.Module):
patch_size = 2 patch_size = 2
f_patch_size = 1 f_patch_size = 1
pH = pW = patch_size pH = pW = patch_size
device = control_context.device
B, C, H, W = control_context.shape B, C, H, W = control_context.shape
control_context = self.control_all_x_embedder[f"{patch_size}-{f_patch_size}"](control_context.view(B, C, H // pH, pH, W // pW, pW).permute(0, 2, 4, 3, 5, 1).flatten(3).flatten(1, 2)) control_context = self.control_all_x_embedder[f"{patch_size}-{f_patch_size}"](control_context.view(B, C, H // pH, pH, W // pW, pW).permute(0, 2, 4, 3, 5, 1).flatten(3).flatten(1, 2))