'update'

2024-07-11 00:54:41 +08:00
parent 661e514c19
commit d73df4cc77
1 changed files with 3 additions and 3 deletions
@@ -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))