fix: continue fix

Author: niushengxiao

lightllm/common/basemodel/layer_weights/meta_weights/mm_weight/colmm_weight.py

@@ -101,7 +101,7 @@ def _load_weights(self, weights: Dict[str, torch.Tensor]) -> None:
             input_scale = weights[self.act_scale_name].to(torch.float)
             self.input_scale = input_scale.cuda(get_current_device_id())
 
-        if self.weight_name in weights and self.weight_scale is not None:
+        if self.weight_scale is not None and isinstance(self.weight, torch.Tensor):
             self.weight = [
                 self.weight,
                 self.weight_scale,

lightllm/common/basemodel/layer_weights/meta_weights/mm_weight/rowmm_weight.py

@@ -130,7 +130,7 @@ def _load_weights(self, weights: Dict[str, torch.Tensor]) -> None:
             input_scale = weights[self.act_scale_name].to(torch.float)
             self.input_scale = input_scale.cuda(get_current_device_id())
 
-        if self.weight_name in weights and self.weight_scale is not None:
+        if self.weight_scale is not None and isinstance(self.weight, torch.Tensor):
             self.weight = [
                 self.weight,
                 self.weight_scale,
@@ -312,7 +312,7 @@ def _load_weights(self, weights: Dict[str, torch.Tensor]) -> None:
             input_scale = weights[self.act_scale_name].to(torch.float)
             self.input_scale = input_scale.cuda(get_current_device_id())
 
-        if self.weight_name in weights and self.weight_scale is not None:
+        if self.weight_scale is not None and isinstance(self.weight, torch.Tensor):
             self.weight = [
                 self.weight,
                 self.weight_scale,