update int_mm minimal pytorch version

huggingface · Nov 12, 2024 · ad9d6f6 · ad9d6f6
1 parent 002ece8
commit ad9d6f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/quanto/library/qbytes_mm.py b/optimum/quanto/library/qbytes_mm.py
@@ -92,7 +92,7 @@ def qbytes_mm_impl_cuda(activations: torch.Tensor, weights: torch.Tensor, output
 def qbytes_mm_impl_cpu(activations: torch.Tensor, weights: torch.Tensor, output_scales: torch.Tensor) -> torch.Tensor:
     if (
         # FIXME: accuracy issues with 2.4.x
-        version.parse(torch.__version__).release > version.parse("2.5.0").release
+        version.parse(torch.__version__).release >= version.parse("2.6.0").release
         and activations.dtype == torch.int8
         and weights.dtype == torch.int8
     ):