kgonia/comfyui-kleinprunedte

ComfyUI-KleinPrunedTE is a small add-on node for ComfyUI, a visual tool for building AI image generation workflows. It adds support for a trimmed-down version of the text encoder used by the FLUX.2-klein-9B image generation model. A text encoder is the part of an image generation system that turns a written prompt into a form the rest of the model can use to create an image. The pruned version of this text encoder has 5.1 billion parameters instead of the original 8.2 billion, which the README says makes it about 35 percent smaller and cuts peak video memory use by around 5 gigabytes during encoding, while producing images that are nearly indistinguishable from the full-size encoder. However, because this pruned version has an irregular internal structure that varies layer by layer, ComfyUI's normal text encoder loading node cannot read it correctly and fails with an error. This project exists specifically to load that pruned checkpoint correctly by figuring out the right shape for each layer directly from the file itself, without needing any extra configuration files. Installation can be done through the ComfyUI Manager by installing from a Git URL, or manually by cloning the repository into ComfyUI's custom_nodes folder and restarting ComfyUI. No additional Python packages are needed since the node only relies on ComfyUI's own built-in tools, though it does require a version of ComfyUI that already supports FLUX.2. Using it requires downloading three separate model files and placing them in specific ComfyUI folders: the pruned text encoder itself, the main FLUX.2-klein-9B image generation model (which is gated and requires accepting a license and logging in on Hugging Face to download), and a matching VAE file. The repository includes an example workflow file that already has the new node wired in correctly with the recommended settings for this particular model. The code for this ComfyUI node is released under the MIT license, but the pruned model weights themselves are a derivative of FLUX.2-klein-9B and carry that model's FLUX Non-Commercial License, meaning the weights cannot be used commercially.