Update gptq.py

auto_gptq/quantization/gptq.py

@@ -113,25 +113,9 @@ class GPTQ:
         damp = percdamp * torch.mean(torch.diag(H))
         diag = torch.arange(self.columns, device=self.dev)
         H[diag, diag] += damp
-        try:
-        
-            H = torch.linalg.cholesky(H)
-            H = torch.cholesky_inverse(H)
-            H = torch.linalg.cholesky(H, upper=True)
-        except torch._C._LinAlgError as e:
-            import pdb
-            pdb.set_trace()
-
-
-            print("Cholesky decomposition failed: ", e)
-            epsilon = 1e-3
-            # 添加一个小的对角线偏移以增强正定性
-            H += epsilon * torch.eye(H.size(0), device=H.get_device())
-
-            H = torch.linalg.cholesky(H)
-            H = torch.cholesky_inverse(H)
-            H = torch.linalg.cholesky(H, upper=True)
-
+        H = torch.linalg.cholesky(H)
+        H = torch.cholesky_inverse(H)
+        H = torch.linalg.cholesky(H, upper=True)
 
         Hinv = H