Fix CI and style (#12658)

2025-11-05 15:08:15 -08:00
parent 80802c4cc6
commit c7d57d5bb3
11 changed files with 96 additions and 70 deletions
@@ -162,6 +162,7 @@ class LinearBase(torch.nn.Module):
        if params_dtype is None:
            params_dtype = torch.get_default_dtype()
        self.params_dtype = params_dtype
+        self.quant_config = quant_config
        if quant_config is None:
            self.quant_method: Optional[QuantizeMethodBase] = UnquantizedLinearMethod()
        else: