|
from functools import cached_property
|
|
from importlib.metadata import version
|
|
from importlib.util import find_spec
|
|
|
|
import torch
|
|
from packaging.version import parse
|
|
|
|
|
|
def get_module_version(module_name):
|
|
try:
|
|
module_version = version(module_name)
|
|
except Exception:
|
|
module_version = None
|
|
return module_version
|
|
|
|
|
|
def compare_version(version1, version2):
|
|
if not isinstance(version1, str) or not isinstance(version2, str):
|
|
return None
|
|
|
|
if parse(version1) > parse(version2):
|
|
return 1
|
|
elif parse(version1) < parse(version2):
|
|
return -1
|
|
else:
|
|
return 0
|
|
|
|
|
|
def compare_module_version(module_name, version_string):
|
|
module_version = get_module_version(module_name)
|
|
|
|
result = compare_version(module_version, version_string)
|
|
return result if result is not None else -2
|
|
|
|
|
|
class IACheckVersions:
|
|
@cached_property
|
|
def diffusers_enable_cpu_offload(self):
|
|
if (find_spec("diffusers") is not None and compare_module_version("diffusers", "0.15.0") >= 0 and
|
|
find_spec("accelerate") is not None and compare_module_version("accelerate", "0.17.0") >= 0 and
|
|
torch.cuda.is_available()):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
@cached_property
|
|
def torch_mps_is_available(self):
|
|
if compare_module_version("torch", "2.0.1") < 0:
|
|
if not getattr(torch, "has_mps", False):
|
|
return False
|
|
try:
|
|
torch.zeros(1).to(torch.device("mps"))
|
|
return True
|
|
except Exception:
|
|
return False
|
|
else:
|
|
return torch.backends.mps.is_available() and torch.backends.mps.is_built()
|
|
|
|
@cached_property
|
|
def torch_on_amd_rocm(self):
|
|
if find_spec("torch") is not None and "rocm" in version("torch"):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
@cached_property
|
|
def gradio_version_is_old(self):
|
|
if find_spec("gradio") is not None and compare_module_version("gradio", "3.34.0") <= 0:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
ia_check_versions = IACheckVersions()
|
|
|