docs(flux2): explain default weight transfers

Author: Watebear

lightx2v/models/networks/flux2/weights/post_weights.py

@@ -35,6 +35,7 @@ def to_cuda(self, non_blocking=True):
         for module in self._modules.values():
             if module is not None and hasattr(module, "to_cuda"):
                 if self.mm_type == "Default":
+                    # Default uses plain transpose views; other layouts need the module's own to_cuda().
                     move_flux2_leaf_to_cuda(module, non_blocking=non_blocking)
                 else:
                     module.to_cuda(non_blocking=non_blocking)

lightx2v/models/networks/flux2/weights/pre_weights.py

@@ -42,6 +42,7 @@ def to_cuda(self, non_blocking=True):
         for module in self._modules.values():
             if module is not None and hasattr(module, "to_cuda"):
                 if self.mm_type == "Default":
+                    # Default uses plain transpose views; other layouts need the module's own to_cuda().
                     move_flux2_leaf_to_cuda(module, non_blocking=non_blocking)
                 else:
                     module.to_cuda(non_blocking=non_blocking)

lightx2v/models/networks/flux2/weights/transformer_weights.py

@@ -214,6 +214,7 @@ def to_cuda(self, non_blocking=True):
         for module in self._modules.values():
             if module is not None and hasattr(module, "to_cuda"):
                 if self.mm_type == "Default":
+                    # Default uses plain transpose views; other layouts need the module's own to_cuda().
                     move_flux2_leaf_to_cuda(module, non_blocking=non_blocking)
                 else:
                     module.to_cuda(non_blocking=non_blocking)