298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-05 06:47:08 +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

6
comfy_extras/nodes_lumina2.py
View File

@ -25,7 +25,7 @@ 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
@ -41,7 +41,7 @@ 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
@ -94,7 +94,7 @@ class CLIPTextEncodeLumina2(ComfyNodeABC):
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {
"CLIPTextEncodeLumina": CLIPTextEncodeLumina2, "CLIPTextEncodeLumina2": CLIPTextEncodeLumina2,
"RenormCFG": RenormCFG "RenormCFG": RenormCFG
} }