wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-26 22:30:19 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Update zluda.py

Browse Source
This commit is contained in:
patientx 2025-05-09 00:17:15 +03:00 committed by GitHub
parent a59deda664
commit 8a3424f354
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
comfy/customzluda/zluda.py
View File

@ -35,48 +35,38 @@ def get_package_version(package_name):
from importlib_metadata import version from importlib_metadata import version
return version(package_name) return version(package_name)
def ensure_package(package_name, required_version, suppress_errors=False): def ensure_package(package_name, required_version):
try: try:
installed_version = get_package_version(package_name) installed_version = get_package_version(package_name)
print(f"Installed version of {package_name}: {installed_version}") print(f"Installed version of {package_name}: {installed_version}")
from packaging import version from packaging import version
if version.parse(installed_version) < version.parse(required_version): if version.parse(installed_version) < version.parse(required_version):
install_package(package_name, required_version, upgrade=True, suppress_errors=suppress_errors) install_package(package_name, required_version, upgrade=True)
print(f"\n{package_name} outdated. Upgraded to {required_version}.") print(f"\n{package_name} outdated. Upgraded to {required_version}.")
except Exception: except Exception:
install_package(package_name, required_version, suppress_errors=suppress_errors) install_package(package_name, required_version)
print(f"\n{package_name} was missing. Installed it.") print(f"\n{package_name} was missing. Installed it.")
def install_package(package_name, version, upgrade=False, suppress_errors=False): def install_package(package_name, version, upgrade=False):
import subprocess import subprocess
import sys import sys
import platform
args = [sys.executable, '-m', 'pip', 'install', f'{package_name}=={version}', '--quiet'] args = [sys.executable, '-m', 'pip', 'install', f'{package_name}=={version}', '--quiet']
if upgrade: if upgrade:
args.append('--upgrade') args.append('--upgrade')
subprocess.check_call(args)
if suppress_errors and platform.system() == 'Windows':
# For Windows with error suppression, we need to join the command and use shell=True
command = ' '.join(args) + ' 2>nul'
subprocess.check_call(command, shell=True)
else:
# Normal execution without shell
subprocess.check_call(args)
# List of packages and their required versions with special flags # List of packages and their required versions
packages_to_check = [ packages_to_check = [
("comfyui-frontend-package", "1.18.9", False), ("comfyui-frontend-package", "1.18.9"),
("comfyui-workflow-templates", "0.1.3", False), ("comfyui-workflow-templates", "0.1.3"),
("av", "14.3.0", True) # Special case: suppress errors for av ("av", "14.3.0")
] ]
# Check and install/update all packages # Check and install/update all packages
for package_name, required_version, suppress_errors in packages_to_check: for package_name, required_version in packages_to_check:
ensure_package(package_name, required_version, suppress_errors) ensure_package(package_name, required_version)
# ------------------- End Version Check ------------------- # ------------------- End Version Check -------------------
# ------------------- Triton Setup ------------------- # ------------------- Triton Setup -------------------
print("\n :: ------------------------ ZLUDA ----------------------- :: ") print("\n :: ------------------------ ZLUDA ----------------------- :: ")
try: try: