nodes/alpertunga-bile-prompt-generator-comfyui
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

quanto torch version checks are updated with newest version

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
generator/__init__.py
+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": if os_name == "Linux":
check_package("bitsandbytes", "bitsandbytes") 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") check_package("optimum-quanto", "optimum-quanto")
generator/utility.py
+1 -1
View File
@@ -38,7 +38,7 @@ class QuantizationType(Enum):
def get_quantization_package() -> QuantizationPackage: def get_quantization_package() -> QuantizationPackage:
if system() == "Linux": if system() == "Linux":
return QuantizationPackage.BITSANDBYTES return QuantizationPackage.BITSANDBYTES
elif check_torch_version_is_enough(2, 2): elif check_torch_version_is_enough(2, 4):
return QuantizationPackage.QUANTO return QuantizationPackage.QUANTO
else: else:
return QuantizationPackage.NONE return QuantizationPackage.NONE