Skip to content

Latest commit

 

History

History
42 lines (32 loc) · 1.82 KB

README.md

File metadata and controls

42 lines (32 loc) · 1.82 KB

Flux Keys Modifier 🧪

This custom node for ComfyUI allows you to dynamically tweak key groups within a FLUX UNet model to change visual rendering styles. Supports FP8, FP16, and FP32 models. The model can be reset between generations or optionally saved.

✨ Features

  • Apply 7 grouped style modifications to UNet weights
  • Individually randomize each group, or randomize all at once
  • Reset model to original weights before each run (optional)
  • Save modified model as .safetensors to the output/ folder
  • Compatible with standard MODEL input/output ports in ComfyUI
  • Designed for FP8/FP16/FP32 models: safe float8 handling (modifies only what can be restored)

⚙️ Parameters

Parameter Description
unet_model UNet model (from Model Loader)
reset_model Reset modified tensors to their original values before applying changes
randomize_all Randomize all sliders (overrides individual toggles)
save_model Save the modified model to disk
save_filename Name of the saved file (output/Flux_keys_modified.safetensors by default)
Keys Group A–G Style sliders (-100 to 200). Adjust rendering behavior
randomize_X Toggle randomization for each slider

💡 Keys Group Legend

These are abstracted tensor categories — not strict visual effects:

  • Keys Group A → qkv.weight
  • Keys Group B → mlp.0.weight
  • Keys Group C → norm.key_norm.scale
  • Keys Group D → attn.proj.bias
  • Keys Group E → img_mlp.0.weight
  • Keys Group F → txt_attn.qkv.weight
  • Keys Group G → time_in.in_layer.bias

🧰 How to Install

Just copy file flux_style_editor_node.py to your ComfyUI custom nodes directory: Then restart ComfyUI.