From 0bf4f88dea43b4a49ce5be06c97dbc7b7bce5888 Mon Sep 17 00:00:00 2001 From: patientx Date: Sat, 8 Mar 2025 14:37:15 +0300 Subject: [PATCH] Update zluda.py --- comfy/zluda.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/comfy/zluda.py b/comfy/zluda.py index bd9a2b718..d6ed044a1 100644 --- a/comfy/zluda.py +++ b/comfy/zluda.py @@ -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):