A modulation layer addon for Flux that reduces model size to 8.8B parameters without significant quality loss.
ComfyUI_FluxMod acts as a plugin for Flux, enabling you to run Flux Dev and Flux Schnell on more consumer-friendly hardware. This is achieved by utilizing a modulation layer that significantly reduces the parameter count while maintaining quality.
Note: You still need a Flux Dev or Flux Schnell model to use this addon.
- Requirements
- Installation
- Usage
- Node Information
- Quantization Guide
- Examples
- FAQ
- Support
- Contributing
- Issues
- Sample Workflow
- ComfyUI installation
- Flux model (Dev or Schnell). Tested to work with the original models, may not work with third party versions such as fine tunes.
- universal_modulator.safetensors
It's recommended to use either ComfyUI Manager or Comfy Registry.
- Install ComfyUI Manager if you haven't already
- Open ComfyUI and click on the Manager Button (puzzle piece icon)
- Go to "Custom Nodes Manager" tab
- Search for "ComfyUI_FluxMod"
- Click Install
- Restart ComfyUI
comfy node registry-install comfyui_fluxmod- Navigate to your ComfyUI's custom_nodes folder
- Clone the repository:
git clone https://github.com/lodestone-rock/ComfyUI_FluxMod.git- Restart ComfyUI
- Refresh your browser if ComfyUI is already running
| Node | Description | Options |
|---|---|---|
| FluxModCheckpointLoader | Primary checkpoint loading node | • ckpt_name: Flux model path • guidance_name: Modulation addon path • quant_mode: Quantization selection |
| KSamplerMod | Modified KSampler for 8-bit quantization | • activation_casting: Switch between bf16 and fp16 |
| FluxModSamplerWrapper | Sampler wrapper for 8-bit quantization | • activation_casting: Switch between bf16 and fp16 |
| SkipLayerForward | Skip specific Flux layers | • skip_mmdit_layers: Which MMDiT layer to skip • skip_dit_layers: Which DIT layers to skip (-1 to disable) |
⚠️ Important: When usingfloat8_e4m3fnorfloat8_e5m2quantization modes, you must use either theKSamplerModnode instead of the regular KSampler orFluxModSamplerWrapperforSamplerCustomor other sampler nodes that take aSAMPLERinput. This requirement does not apply when usingbf16mode inFluxModCheckpointLoaderor when starting ComfyUI with--fast.
- Double click workspace → search "FluxModCheckpointLoader"
- Select your Flux model in
ckpt_name - Select modulation addon in
guidance_name - Choose quantization mode
- Configure remaining nodes as per standard Flux workflow
💡 Tip: Check the
examplesfolder for sample workflows. Drag and drop the workflow image into ComfyUI to get started quickly.
| Mode | Recommended GPU | VRAM Usage | Recommended |
|---|---|---|---|
| bf16 | 24GB+ | ~19GB | ✅ |
| float8_e4m3fn | 12-16GB | ~10GB | ✅ |
| float8_e5m2 | 12-16GB | ~10GB | ❌ |
Here are some comparison examples showing the output quality between the original Flux model and FluxMod:
Example 1: Art Studio Scene
Prompt: A photo of an art studio with a cabin design, there are paint splatters over much of the wooden furnishing, there are old style windows overlooking a lake outside with a slanted ceiling with large skylights letting in natural light. There is an easel with a half-finished painting. There is a paint palette that is placed on a wooden table. There are dust speckles floating in the air illuminated by the golden hour light. There is a woman standing in a colourful dress and ruby red shoes who is painting in front of the easel.
Example 2: Glass Rainbow Room
Prompt: A photo of a room with the walls made of millions of pieces of shattered glass in rainbow colours, there is light shining through the glass causing a huge dispersion of colours across the scene. The light is refracting off all the glass in the room illuminating the room with bright hues from the colored glass.
Example 3: Balloon Text
Prompt: Text made out of foil balloons saying "Hi there fellow traveller, make sure to star this GitHub! Thank you!"
Example 4: Space Pirate Ship
Prompt: Concept art of a ornately decorated pirate ship in outer space that is floating through a nebula with spectacular blue and purple hues. There is dust that is around the ship as it sails through the cosmos, dispersing at the bow of the ship.
Example 5: Detailed Room Scene
Prompt: A scene with a blue block and a red ball that is placed on top of the blue block. In the background there is a green door with pink walls. To the left of the door, there is a painting on the wall which is showing a scene of an ocean wave that is orange. To the right of the door, there is a portrait of a cat that has purple eyes. There is a window to the right side of the scene that is letting in light that has a shade of blue, illuminating the carpet which has a cyan hue. There is a bed on the left side of the room that has a white pillow with orange sheets. There is a bedside table with an orange lamp and a phone that is placed on top of it. To the right side of the beside table, there is a green bin with a red recycling logo on it.
Will my outputs be different?
Yes, outputs will likely differ as we're reducing parameters. However, the difference is often minimal for most use cases.How much does this degrade image quality?
Testing shows minimal quality degradation in most cases. The most notable exception is long text generation, which shows a moderate degradation.Why use this over regular quantization?
You don't have to use this over regular quantisation! You can combine them, or if you don't want to use quantisation at all and you have enough VRAM, you can also just stick with bf16. If you combine quantisation, you can make the model even smaller and allow it to run on consumer hardware.I tried exporting/saving this model and I got an error?
This model has a completely different architecture compared to the original Flux and none of the current methods for exporting/saving models would support it. This is why we needed to have this custom node created in the first place, since otherwise it wouldn't load properly.Need help? Join our Discord community for support and discussions.
Pull requests are welcome! Feel free to contribute to the project by:
- Fixing bugs
- Adding new features
- Improving documentation
- Suggesting enhancements
Found a bug? Have a suggestion? Please create a GitHub issue with:
- A clear description of the problem
- Steps to reproduce
- Expected vs actual behavior
- Your environment details (OS, GPU, etc.)
