wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-19 02:40:15 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Update zluda.py

Browse Source
This commit is contained in:
patientx 2025-05-08 19:57:59 +03:00 committed by GitHub
parent f9671afff0
commit 81a16eefbc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
comfy/zluda.py
View File

@ -63,6 +63,26 @@ except Exception:
import sys import sys
subprocess.check_call([sys.executable, '-m', 'pip', 'install', f'{templates_package_name}=={required_templates_version}', '--quiet']) subprocess.check_call([sys.executable, '-m', 'pip', 'install', f'{templates_package_name}=={required_templates_version}', '--quiet'])
print("\nComfyUI Workflow Templates was missing. Installed it.") print("\nComfyUI Workflow Templates was missing. Installed it.")
# Check and install/update pyav
required_templates_version = "14.3.0"
templates_package_name = "av"
try:
installed_version = get_package_version(templates_package_name)
print(f"Installed version of {templates_package_name}: {installed_version}")
from packaging import version
if version.parse(installed_version) < version.parse(required_templates_version):
import subprocess
import sys
subprocess.check_call([sys.executable, '-m', 'pip', 'install', f'{templates_package_name}=={required_templates_version}', '--quiet', '--upgrade'])
print(f"\nPYAV outdated. Upgraded to {required_templates_version}.")
except Exception:
import subprocess
import sys
subprocess.check_call([sys.executable, '-m', 'pip', 'install', f'{templates_package_name}=={required_templates_version}', '--quiet'])
print("\nPYAV was missing. Installed it.")
# ------------------- End Frontend Version Check ------------------- # ------------------- End Frontend Version Check -------------------
# ------------------- ZLUDA Detection ------------------- # ------------------- ZLUDA Detection -------------------