wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-12 15:20:51 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Update zluda.py

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

10
comfy/zluda.py
View File

@ -16,10 +16,18 @@ os.environ['DISABLE_ADDMM_CUDA_LT'] = '1'
import torch
# Ensure pkg_resources is installed
try:
import pkg_resources
except ImportError:
import subprocess
import sys
subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'setuptools', '--quiet'])
import pkg_resources
# 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):