fix sparge bugs

lightx2v/common/ops/attn/attn_weight.py

@@ -58,6 +58,11 @@ class AttnWeightTemplate(metaclass=ABCMeta):
     def to_cuda(self, non_blocking=False):
         self.weight = self.weight.cuda(non_blocking=non_blocking)
 
+    def state_dict(self, destination=None):
+        if destination is None:
+            destination = {}
+        return destination
+
 
 @ATTN_WEIGHT_REGISTER("flash_attn2")
 class FlashAttn2Weight(AttnWeightTemplate):

lightx2v/utils/set_config.py

@@ -20,6 +20,7 @@ def get_default_config():
         "strength_model": 1.0,
         "mm_config": {},
         "use_prompt_enhancer": False,
+        "sparge": False,
     }
     return default_config