ia_check_versions.py

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()