wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-22 12:30:50 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

resolved Ruff lint errors

Browse Source
This commit is contained in:
nolan4 2025-10-28 19:05:34 -07:00
parent 79c30e1630
commit 2d550102fc
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/ldm/qwen_image/model.py
View File

@ -2,7 +2,6 @@
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
import math
import logging import logging
from typing import Optional, Tuple from typing import Optional, Tuple
from einops import repeat, rearrange from einops import repeat, rearrange
@ -336,7 +335,6 @@ class QwenImageTransformer2DModel(nn.Module):
self.inner_dim = num_attention_heads * attention_head_dim self.inner_dim = num_attention_heads * attention_head_dim
self.pe_embedder = EmbedND(dim=attention_head_dim, theta=10000, axes_dim=list(axes_dims_rope)) self.pe_embedder = EmbedND(dim=attention_head_dim, theta=10000, axes_dim=list(axes_dims_rope))
self.time_text_embed = QwenTimestepProjEmbeddings( self.time_text_embed = QwenTimestepProjEmbeddings(
embedding_dim=self.inner_dim, embedding_dim=self.inner_dim,
pooled_projection_dim=pooled_projection_dim, pooled_projection_dim=pooled_projection_dim,
@ -560,8 +558,8 @@ class QwenImageTransformer2DModel(nn.Module):
# Positive gets entity constraints, negative gets standard attention (all zeros) # Positive gets entity constraints, negative gets standard attention (all zeros)
logger.debug( logger.debug(
f"[EliGen Model] CFG batched detected - creating separate masks. " "[EliGen Model] CFG batched detected - creating separate masks. "
f"Positive (index 0) gets entity mask, Negative (index 1) gets standard mask" "Positive (index 0) gets entity mask, Negative (index 1) gets standard mask"
) )
# Create standard attention mask (all zeros = no constraints) # Create standard attention mask (all zeros = no constraints)