add Flux_ControlNet_Quantization

2026-03-22 00:38:11 +00:00 · 2024-10-29 17:29:24 +08:00
parent 7e97a96840
commit 900a1c095f
7 changed files with 558 additions and 2 deletions
--- a/diffsynth/models/flux_dit.py
+++ b/diffsynth/models/flux_dit.py
@@ -475,6 +475,9 @@ class FluxDiT(torch.nn.Module):
                    # del module
                    setattr(model, name, new_layer)
                elif isinstance(module, RMSNorm):
+                    if hasattr(module,"quantized"):
+                        continue
+                    module.quantized= True
                    new_layer = quantized_layer.RMSNorm(module)
                    setattr(model, name, new_layer)
                else: