From 55e8737dc5de7558e065e2ea9766e3945837b342 Mon Sep 17 00:00:00 2001 From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com> Date: Tue, 2 Dec 2025 21:29:26 -0500 Subject: [PATCH] Fix ruff. --- comfy/ldm/lumina/controlnet.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/comfy/ldm/lumina/controlnet.py b/comfy/ldm/lumina/controlnet.py index 8246d5fde..fd7ce3b5c 100644 --- a/comfy/ldm/lumina/controlnet.py +++ b/comfy/ldm/lumina/controlnet.py @@ -1,6 +1,5 @@ import torch from torch import nn -import math from .model import JointTransformerBlock @@ -102,7 +101,6 @@ class ZImage_Control(torch.nn.Module): patch_size = 2 f_patch_size = 1 pH = pW = patch_size - device = control_context.device 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))