Fix weight loading without quant

lightx2v/common/ops/mm/mm_weight.py

@@ -64,7 +64,7 @@ class MMWeight(MMWeightTemplate):
         super().__init__(weight_name, bias_name)
 
     def load(self, weight_dict):
-        if GET_RUNNING_FLAG() == "save_naive_quant" or self.config.get("weight_auto_quant", False):
+        if GET_RUNNING_FLAG() == "save_naive_quant" or self.config.get("weight_auto_quant", False) or self.config.get("mm_type", "Default") == "Default":
             self.weight = weight_dict[self.weight_name].t().cuda()
             self.bias = weight_dict[self.bias_name].cuda() if self.bias_name is not None else None
         else:

lightx2v/models/networks/hunyuan/model.py

@@ -73,7 +73,7 @@ class HunyuanModel:
         return weight_dict
 
     def _init_weights(self):
-        if GET_RUNNING_FLAG() == "save_naive_quant" or self.config["mm_config"].get("weight_auto_quant", False):
+        if GET_RUNNING_FLAG() == "save_naive_quant" or self.config["mm_config"].get("weight_auto_quant", False) or self.config["mm_config"].get("mm_type", "Default") == "Default":
             weight_dict = self._load_ckpt()
         else:
             weight_dict = self._load_ckpt_quant_model()

lightx2v/models/networks/wan/model.py

@@ -90,7 +90,7 @@ class WanModel:
 
     def _init_weights(self, weight_dict=None):
         if weight_dict is None:
-            if GET_RUNNING_FLAG() == "save_naive_quant" or self.config["mm_config"].get("weight_auto_quant", False):
+            if GET_RUNNING_FLAG() == "save_naive_quant" or self.config["mm_config"].get("weight_auto_quant", False) or self.config["mm_config"].get("mm_type", "Default") == "Default":
                 self.original_weight_dict = self._load_ckpt()
             else:
                 self.original_weight_dict = self._load_ckpt_quant_model()