🎨 FLUX.2 [klein] 4B AIO Workflow v1

UPDATE:

Thanks again for all the feedback! I wanted to give an update regarding the issue some of you have reported. When using the FP8 AIO model, some users ran into a CLIPTextEncode error, saying the clip input was invalid or the text encoder couldn’t be recognized.

I’ve realized that this might be caused by the script I wrote, which probably doesn’t set the right parameters for ComfyUI’s AIO version. I’m not entirely sure why it worked on my version, but to avoid confusion for everyone, I’ve deleted the current upload on Civitai.

I’ll redownload ComfyUI without any custom nodes tonight and try to reproduce and fix the problem. Once everything is working correctly, I’ll re-upload the model. Thanks for your patience!

UPDATE 2:

After extensive debugging, I found the root cause of the issue - and it's NOT the model files!

The Problem:

The AIO models are correctly formatted. The issue is that ComfyUI's Flux2 class has the clip_target() method

returning None (marked as "TODO" in the code). This means ComfyUI currently does NOT support loading text encoders

from Flux2 AIO checkpoints - regardless of how the model is formatted.

When you load a Flux2 AIO checkpoint:

- ✅ UNET loads correctly

- ✅ VAE loads correctly

- ❌ Text Encoder is completely ignored (because of the missing implementation)

This is a ComfyUI limitation, not a model issue.

The relevant code in comfy/supported_models.py (line ~772):

```python

class Flux2(Flux):

def clip_target(self, state_dict={}):

return None # TODO ← This is the problem

Workaround (until ComfyUI fixes this):

You need to modify ComfyUI/comfy/supported_models.py and replace the clip_target() method in the Flux2 class. I've

submitted a GitHub issue with the fix:

👉 https://github.com/Comfy-Org/ComfyUI/issues/12032

Once ComfyUI implements this, the AIO models will work out of the box with the standard "Load Checkpoint" node.

I'll re-upload the models once this is resolved or provide a detailed guide on how to apply the fix manually. Sorry

for the confusion - I didn't realize my local ComfyUI had this method already implemented! because I wanted to create the AIO versions with comfyui first

UPDATE 3:

There are currently three ways to use the AIO version. It turns out that the files themselves are completely fine — the actual issue is that ComfyUI does not recognize the text encoder in certain versions.

Fix 1: Follow the instructions from UPDATE 2.
Fix 2: Update ComfyUI to the newer beta version from 2026-01-22, where this issue has already been fixed.
Fix 3: Wait for the next stable ComfyUI release after v0.10.0, which should also include this fix.

Thanks again to everyone for the great feedback. I apologize for any inconvenience this may have caused, and I hope you still have lots of fun generating.
May the AI be with you 😊