298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-06 15:27:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix issues with the ruff style check

Browse Source
This commit is contained in:
lzyhha 2025-02-16 11:43:04 +00:00
parent 7ed909f457
commit 1c4bbc2c53
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
comfy_extras/nodes_lumina2.py
View File

@ -25,11 +25,11 @@ class RenormCFG:
if timestep[0] < cfg_trunc: if timestep[0] < cfg_trunc:
cond_eps, uncond_eps = cond_denoised[:, :in_channels], uncond_denoised[:, :in_channels] cond_eps, uncond_eps = cond_denoised[:, :in_channels], uncond_denoised[:, :in_channels]
cond_rest, uncond_rest = cond_denoised[:, in_channels:], uncond_denoised[:, in_channels:] cond_rest, _ = cond_denoised[:, in_channels:], uncond_denoised[:, in_channels:]
half_eps = uncond_eps + cond_scale * (cond_eps - uncond_eps) half_eps = uncond_eps + cond_scale * (cond_eps - uncond_eps)
half_rest = cond_rest half_rest = cond_rest
if float(renorm_cfg) > 0.0: if float(renorm_cfg) > 0.0:
ori_pos_norm = torch.linalg.vector_norm(cond_eps ori_pos_norm = torch.linalg.vector_norm(cond_eps
, dim=tuple(range(1, len(cond_eps.shape))), keepdim=True , dim=tuple(range(1, len(cond_eps.shape))), keepdim=True
) )
@ -41,10 +41,10 @@ class RenormCFG:
half_eps = half_eps * (max_new_norm / new_pos_norm) half_eps = half_eps * (max_new_norm / new_pos_norm)
else: else:
cond_eps, uncond_eps = cond_denoised[:, :in_channels], uncond_denoised[:, :in_channels] cond_eps, uncond_eps = cond_denoised[:, :in_channels], uncond_denoised[:, :in_channels]
cond_rest, uncond_rest = cond_denoised[:, in_channels:], uncond_denoised[:, in_channels:] cond_rest, _ = cond_denoised[:, in_channels:], uncond_denoised[:, in_channels:]
half_eps = cond_eps half_eps = cond_eps
half_rest = cond_rest half_rest = cond_rest
cfg_result = torch.cat([half_eps, half_rest], dim=1) cfg_result = torch.cat([half_eps, half_rest], dim=1)
# cfg_result = uncond_denoised + (cond_denoised - uncond_denoised) * cond_scale # cfg_result = uncond_denoised + (cond_denoised - uncond_denoised) * cond_scale
@ -59,7 +59,7 @@ class RenormCFG:
class CLIPTextEncodeLumina2(ComfyNodeABC): class CLIPTextEncodeLumina2(ComfyNodeABC):
SYSTEM_PROMPT = { SYSTEM_PROMPT = {
"superior": "You are an assistant designed to generate superior images with the superior "\ "superior": "You are an assistant designed to generate superior images with the superior "\
"degree of image-text alignment based on textual prompts or user prompts.", "degree of image-text alignment based on textual prompts or user prompts.",
"alignment": "You are an assistant designed to generate high-quality images with the "\ "alignment": "You are an assistant designed to generate high-quality images with the "\
"highest degree of image-text alignment based on textual prompts." "highest degree of image-text alignment based on textual prompts."
} }
@ -94,7 +94,7 @@ class CLIPTextEncodeLumina2(ComfyNodeABC):
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {
"CLIPTextEncodeLumina": CLIPTextEncodeLumina2, "CLIPTextEncodeLumina2": CLIPTextEncodeLumina2,
"RenormCFG": RenormCFG "RenormCFG": RenormCFG
} }