add Flux_ControlNet_Quantization

2024-10-29 17:29:24 +08:00
parent 7e97a96840
commit 900a1c095f
7 changed files with 558 additions and 2 deletions
--- a/diffsynth/controlnets/controlnet_unit.py
+++ b/diffsynth/controlnets/controlnet_unit.py
@@ -31,6 +31,8 @@ class MultiControlNetManager:
    def to(self, device):
        for model in self.models:
            model.to(device)
+        for processor in self.processors:
+            processor.to(device)
    
    def process_image(self, image, processor_id=None):
        if processor_id is None: