quanto torch version checks are updated with newest version

This commit is contained in:
alpertunga-bile
2024-09-20 22:27:50 +03:00
parent 72475ef839
commit 6c7a368d79
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -58,5 +58,5 @@ def check_torch_version_is_enough(min_major: int, min_minor: int) -> bool:
if os_name == "Linux":
check_package("bitsandbytes", "bitsandbytes")
elif check_torch_version_is_enough(2, 2):
elif check_torch_version_is_enough(2, 4):
check_package("optimum-quanto", "optimum-quanto")
+1 -1
View File
@@ -38,7 +38,7 @@ class QuantizationType(Enum):
def get_quantization_package() -> QuantizationPackage:
if system() == "Linux":
return QuantizationPackage.BITSANDBYTES
elif check_torch_version_is_enough(2, 2):
elif check_torch_version_is_enough(2, 4):
return QuantizationPackage.QUANTO
else:
return QuantizationPackage.NONE