diff --git a/chatglm3/quantization.py b/chatglm3/quantization.py index aa3be50..a6d74d2 100644 --- a/chatglm3/quantization.py +++ b/chatglm3/quantization.py @@ -148,11 +148,11 @@ class QuantizedLinear(torch.nn.Module): try: self.weight = Parameter( - self.weight.to(device), requires_grad=False) + self.weight.to(device)) self.weight_scale = Parameter( - self.weight_scale.to(device), requires_grad=False) + self.weight_scale.to(device)) self.bias = Parameter( - bias.to(device), requires_grad=False) if bias is not None else None + bias.to(device)) if bias is not None else None except Exception as e: self.weight = Parameter(self.weight.to(device)) self.weight_scale = Parameter(self.weight_scale.to(device))