From 08381b1a9d037ec9aa33f5943bfa0414d78f1e22 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Mon, 22 Jun 2026 21:20:12 +0300 Subject: [PATCH] Update model.py --- comfy/ldm/krea2/model.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/comfy/ldm/krea2/model.py b/comfy/ldm/krea2/model.py index e8cdeed34..ecb16254f 100644 --- a/comfy/ldm/krea2/model.py +++ b/comfy/ldm/krea2/model.py @@ -1,10 +1,8 @@ """Krea 2 (K2) — single-stream MMDiT. -Native ComfyUI port of Krea's reference ``mmdit.py`` (``SingleStreamDiT``). Text tokens -(produced by a Qwen3-VL-4B 12-layer ``txtfusion`` adapter) and patchified image tokens are +Text tokens produced by a Qwen3-VL-4B 12-layer ``txtfusion`` adapter and patchified image tokens are concatenated into one sequence and run through ``layers`` shared transformer blocks with -AdaLN-single modulation, GQA + per-head QK-norm + sigmoid-gated attention, SwiGLU MLP, and -3-axis RoPE. +AdaLN-single modulation, GQA + per-head QK-norm + sigmoid-gated attention, SwiGLU MLP, and 3-axis RoPE. """ from typing import Optional