wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-03 21:20:49 +08:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

Update model.py

Browse Source
This commit is contained in:
kijai 2026-06-22 21:20:12 +03:00
parent 8f3e50b8b7
commit 08381b1a9d
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/krea2/model.py
View File

@ -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