wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-22 04:10:15 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

updated how to handle comfyui package updates

Browse Source
This commit is contained in:
patientx 2025-05-13 16:22:44 +03:00 committed by GitHub
parent f0127d6326
commit 6fc773788f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 33 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
comfy/zluda.py
View File

@ -24,6 +24,24 @@ 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 parse_requirements_file(requirements_path):
"""Parse requirements.txt file and extract package versions."""
requirements = {}
try:
with open(requirements_path, 'r') as f:
for line in f:
line = line.strip()
if line and not line.startswith('#'):
if '==' in line:
pkg, version = line.split('==', 1)
requirements[pkg] = version.strip()
elif '>=' in line:
pkg, version = line.split('>=', 1)
requirements[pkg] = version.strip()
except FileNotFoundError:
print(f" :: Warning: requirements.txt not found at {requirements_path}")
return requirements
def ensure_package(package_name, required_version): def ensure_package(package_name, required_version):
try: try:
installed_version = get_package_version(package_name) installed_version = get_package_version(package_name)
@ -40,21 +58,27 @@ def ensure_package(package_name, required_version):
def install_package(package_name, version, upgrade=False): def install_package(package_name, version, upgrade=False):
import subprocess import subprocess
import sys import sys
args = [sys.executable, '-m', 'pip', 'install', f'{package_name}=={version}', '--quiet'] args = [sys.executable, '-m', 'pip', 'install',
f'{package_name}=={version}',
'--quiet',
'--disable-pip-version-check']
if upgrade: if upgrade:
args.append('--upgrade') args.append('--upgrade')
subprocess.check_call(args) subprocess.check_call(args)
# List of packages and their required versions import os
packages_to_check = [ requirements_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'requirements.txt')
("comfyui-frontend-package", "1.19.9"), required_packages = parse_requirements_file(requirements_path)
("comfyui-workflow-templates", "0.1.14"),
("av", "14.3.0") packages_to_monitor = [
"comfyui-frontend-package",
"comfyui-workflow-templates",
"av"
] ]
# Check and install/update all packages for package_name in packages_to_monitor:
for package_name, required_version in packages_to_check: if package_name in required_packages:
ensure_package(package_name, required_version) ensure_package(package_name, required_packages[package_name])
# ------------------- End Version Check ------------------- # ------------------- End Version Check -------------------
# ------------------- ZLUDA Detection ------------------- # ------------------- ZLUDA Detection -------------------