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

Update zluda.py

Browse Source
This commit is contained in:
patientx 2025-03-08 14:33:15 +03:00 committed by GitHub
parent 0f7de5c588
commit 2ce9177547
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
comfy/zluda.py
View File

@ -16,15 +16,24 @@ os.environ['DISABLE_ADDMM_CUDA_LT'] = '1'
import torch
# Check and install comfyui-frontend-package if not installed
# Check and install comfyui-frontend-package if not installed or if the version is lower than required
required_version = "1.11.8"
try:
import pkg_resources
import comfyui_frontend_package
installed_version = pkg_resources.get_distribution("comfyui-frontend-package").version
if pkg_resources.parse_version(installed_version) < pkg_resources.parse_version(required_version):
import subprocess
import sys
subprocess.check_call([sys.executable, '-m', 'pip', 'install', f'comfyui-frontend-package=={required_version}', '--quiet', '--upgrade'])
print(" ")
print(f"Comfyui Frontend Package version {installed_version} is outdated, updating to version {required_version}. (one time only)")
except ImportError:
import subprocess
import sys
subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'comfyui-frontend-package==1.11.8', '--quiet'])
subprocess.check_call([sys.executable, '-m', 'pip', 'install', f'comfyui-frontend-package=={required_version}', '--quiet'])
print(" ")
print("Comfyui Frontend Package missing / older version , it is installed. (one time only) ")
print("Comfyui Frontend Package missing, it is installed. (one time only) ")
#audio patch
import torch._dynamo