reference unet

modules/ref_controller.py

@@ -17,23 +17,28 @@ class RefController:
         self.modules.append(module)
 
     def set_mode(self, mode: RefMode):
+        print('------- CTRL SETTING MODE TO', mode, 'FOR', len(self.modules))
         for module in self.modules:
             module.ref_mode = mode
 
     def clear_modules(self):
+        print('------- CTRL CLEARING MODULES FOR', len(self.modules))
         for module in self.modules:
             module.ref_bank = None
         self.set_mode(RefMode.OFF)
 
     def set_cfg(self, cfg: bool):
+        print('------- CTRL SETTING CFG TO', cfg, 'FOR', len(self.modules))
         for module in self.modules:
             module.is_cfg = cfg
 
     def set_attention_bound(self, attention_bound: float):
+        print('------- CTRL SETTING attention bound  TO', attention_bound, 'FOR', len(self.modules))
         for module in self.modules:
             module.attention_bound = attention_bound
 
-    def set_style_fidelity(self, style_fidelity: float):
+    def set_style_fidelity(self, style_fidelity: float): 
+        print('------- CTRL SETTING style fidelity  TO', style_fidelity, 'FOR', len(self.modules))
         for module in self.modules:
             module.style_fidelity = style_fidelity
 
@@ -51,5 +56,4 @@ class RefController:
             module.video_length = video_length
 
     def set_temporal_values_to_default(self):
-        self.set_temporal_values_to(
-            self.default_full_length, self.default_video_length)
+        self.set_temporal_values_to(self.default_full_length, self.default_video_length)