diff --git a/moss_nodes.py b/moss_nodes.py index b3ec9d6..ee8d4b0 100644 --- a/moss_nodes.py +++ b/moss_nodes.py @@ -156,11 +156,11 @@ class PROMPTFORGE_KreaEditModelPatch: common = dict(source_latent=source_latent, source_latent_b=source_latent_b, ref_boost=ref_boost, ref_boost_a=ref_boost_a, fit_mode=fit_mode, vae=vae, source_image=source_image, target_latent=target_latent) + rest = {key: value for key, value in common.items() if key != "source_latent"} if use_nag and nag_negative is not None: - rest = {key: value for key, value in common.items() if key != "source_latent"} return comfy_nodes.NODE_CLASS_MAPPINGS["Krea2EditNormalizedAttentionGuidance"]().patch( model, nag_negative, source_latent, phi, tau, alpha, 1000.0, 0.0, **rest) - return comfy_nodes.NODE_CLASS_MAPPINGS["Krea2EditModelPatch"]().patch(model, source_latent, **common) + return comfy_nodes.NODE_CLASS_MAPPINGS["Krea2EditModelPatch"]().patch(model, source_latent, **rest) class PROMPTFORGE_SkipLayerGuidance: diff --git a/pyproject.toml b/pyproject.toml index c584244..32f602f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "ComfyUI-PromptForge" description = "ComfyUI prompt enhancement suite: model/config node with GPU-aware auto settings, prompt enhancer with a preset catalog, and a standalone config finder." -version = "1.28.19" +version = "1.28.20" license = { file = "LICENSE" } dependencies = [ "transformers",