**"Turbo or base CFG?"** Turbo. CFG 1.0, 8 steps.
**"Why does my negative prompt do nothing?"** Because CFG is 1.0. Negative
prompts require classifier-free guidance, which a distilled model doesn't run.
This isn't a bug and there's no setting to change.
**"Can I use my Illustrious/Pony prompts?"** Not directly. Rewrite them as
sentences. Booru tags are a vocabulary this model was never trained on.
**"Why is it slow?"** Cost scales with total tokens, and text and image share
one attention sequence. 832×1216 is roughly 2x the cost of 624×912.
**CFG 1.0. Not 3.5, not 7.** This is guidance-distilled. Raising CFG does not
make it follow the prompt harder — it runs the transformer twice per step
(so ~2x the time) and blows out contrast. If your images look burnt or
over-saturated, CFG is the first thing to check.
| setting | value |
|---|---|
| **CFG** | **1.0** |
| **Steps** | **8** (4–8 all work; 8 is the sweet spot) |
| **Sampler** | Euler — flow-matching (`FlowMatchEulerDiscreteScheduler`) |
| **Scheduler / sigmas** | linear / normal |
| **Shift** | 3.1582 (Krea 2's own default — leave it alone) |
| **Negative prompt** | **does nothing.** Inert at CFG 1.0. Don't bother writing one. |
| **Resolution** | 832×1216, 1216×832, or 1024×1024 |
There is only one sampler for this architecture. Krea 2 is a rectified-flow
model — Euler a, DPM++ 2M, UniPC and friends solve a different equation and
simply aren't options. If your UI shows a sampler list, pick Euler.
**On the sigma scheduler:** I ran a blind 4-way test (linear / karras /
exponential / beta, two seeds). The curves measurably differ but the difference
is not visible — blind rankings came out anti-correlated between seeds. Use
linear and don't think about it again.Description
FAQ
Comments (38)
Run this as a turbo model or base model CFG? Also First! I know you haven't had time to upload the particulars yet .
Turbo
Getting and error on load:
# ComfyUI Error Report ## Error Details - Node ID: 173:165 - Node Type: UNETLoader - Exception Type: RuntimeError - Exception Message: RuntimeError: Error(s) in loading state_dict for SingleStreamDiT: size mismatch for blocks.0.mod.lin: copying a param with shape torch.Size([6, 6144]) from checkpoint, the shape in current model is torch.Size([36864]).
I'm using: qwen3vl_4b_fp8 text_encode and Qwent_image_vae. Is this set-up for something different?
i'm getting the torch.size 6,6144 error too.
on latest forge neo and used krea2realvae_v10 and qwen3vl_4b_bf16
\Stable Diffusion WebUI Forge - Neo\venv\Lib\site-packages\torch\nn\modules\module.py", line 2639, in load_state_dict
raise RuntimeError(
...<3 lines>...
)
RuntimeError: Error(s) in loading state_dict for SingleStreamDiT:
size mismatch for blocks.0.mod.lin: copying a param with shape torch.Size([6, 6144]) from checkpoint, the shape in current model is torch.Size([36864]).
Exception Message: RuntimeError: Error(s) in loading state_dict for SingleStreamDiT: size mismatch for blocks.0.mod.lin: copying a param with shape torch.Size([6, 6144]) from checkpoint, the shape in current model is torch.Size([36864]). size mismatch for blocks.1.mod.lin: copying a param with shape torch.Size([6, 6144]) from checkpoint, the shape in current model is torch.Size([36864]). same error
yeah im getting errors in comfyui too.
Same error here.
Same here:
# ComfyUI Error Report ## Error Details - Node ID: 82:68 - Node Type: UNETLoader - Exception Type: RuntimeError - Exception Message: RuntimeError: Error(s) in loading state_dict for SingleStreamDiT: size mismatch for blocks.0.mod.lin: copying a param with shape torch.Size([6, 6144]) from checkpoint, the shape in current model is torch.Size([36864]).
I have the same problem as you. However, the author provides no explanation.
▎ Confirmed — that's a bug in how I converted the file to single-file format, not your setup. Your text encoder and VAE are correct. The modulation tensors got written in the wrong shape. Fix is built, uploading a corrected version shortly. Sorry for the trouble, and thanks for the clear
▎ error report — it named the exact tensor, which made it quick to find.
fix is uploaded
@bonkaiii187 Thank you for the update. :) This model looks really good. Do you have any plans to release a fp8 version?
@belember i can if people want it
@bonkaiii187 Don't bother with FP8; int8 is fine. Takes all of 15 seconds using the comfy tools sscript.
@bonkaiii187 Hmm, I am still getting an error:
Possible reason: ComfyUI execution error (Node 4: UNETLoader): Error(s) in loading state_dict for SingleStreamDiT:
size mismatch for last.modulation.lin: copying a param with shape torch.Size([12288]) from checkpoint, the shape in current model is torch.Size([2, 6144]).
I get errors whether the model is in the diffusion or stable diffusion folders
@clueless_engineer Do you know if an int8 will run on an AMD video card? I know fp8 does.
@bonkaiii187 An fp8? Yes, please.
@belember @bonkaiii187 Yep still same shape error. I'm getting the error @belember listed
uploading a 2nd fixed version now and i just added an fp8 version as well
@bonkaiii187 Thank you, downloading ...
@bonkaiii187 Thank you! Grabbing it now...
@bonkaiii187 Latest fix worked for me. Thanks so much for the amazing model!
Thanks, now for me is working :) , great one
@bonkaiii187 I didn't feel like this needed it's comment, but no matter what vae ,text encoder or res i use i get this weird diamond cross stitching effect on everything. it goes from subtle to very obvious depending on the surface. not sure if this is a forge neo thing, or somehthing, but thought i would bring it up.
@Marukyu9 haven't heard anyone else say anything about it i would guess it's forge neo. but i know what could cause that type of thing if it's the model
@bonkaiii187 okay, i think i figured it out, i was using --pynvml , removing it cleaned it up mostly.
Still getting errors with the fixed model :(
RuntimeError: Error(s) in loading state_dict for SingleStreamDiT:
size mismatch for last.modulation.lin: copying a param with shape torch.Size([12288]) from checkpoint, the shape in current model is torch.Size([2, 6144]).
model.diffusion_model.last.modulation.lin is what needs changed.
weird i'll try a new version
uploading a 2nd fixed version now and i just added an fp8 version as well
@bonkaiii187 Hi, I tried the fp8 version and I get more errors. I tried moving the model from and back again from the diffusion to the stable_diffusion folders to see if it is a location issue (I use SwarmUI)
ComfyUI execution error (Node 4: CheckpointLoaderSimple): Error(s) in loading state_dict for SingleStreamDiT:
size mismatch for last.modulation.lin: copying a param with shape torch.Size([12288]) from checkpoint, the shape in curr
ent model is torch.Size([2, 6144]).
@belember i'm using mps so i can't test the fp8 versions but if you keep giving me the errors i can fix it. i'm uploading a fix now
@bonkaiii187 2nd fix worked for me. the Fp8 did not.
@bonkaiii187 The new fp8 version is working!!!! yeah!!!! Thank you :)
Third times the charm lol. Working great now, thanks for the excellent model!
@bonkaiii187 working for me now. :D thank you!
@bonkaiii187 Your older FP8 looks like it's still up, not sure if that's intentional. One has "fixed" at the end of the filename though.
@TrashWaifuIsBestWaifu cleaned it up :)