diff --git a/install.py b/install.py index fa6ba13..75b1f84 100644 --- a/install.py +++ b/install.py @@ -39,6 +39,15 @@ def get_pytorch_version(): return None +def check_xformers(): + try: + import xformers + + return True + except ImportError: + return False + + def get_pytorch_cuda_version(): """Extract CUDA version from PyTorch.""" try: @@ -127,6 +136,11 @@ def main(): # Step 2: Check xformers compatibility print("\n[2/3] Checking xformers compatibility...") + + if check_xformers(): + print("xformers exists skip installation") + return 0 + pytorch_version = get_pytorch_version() cuda_version = get_pytorch_cuda_version() diff --git a/pyproject.toml b/pyproject.toml index 8d221f0..4efc7d4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui-starwrapper" description = "A ComfyUI custom node wrapper for [STAR (Spatial-Temporal Augmentation with Text-to-Video Models for Real-World Video Super-Resolution)](https://github.com/NJU-PCALab/STAR), enabling high-quality video upscaling with AI-powered super-resolution." -version = "0.1.3" +version = "0.1.4" license = {file = "LICENSE"} classifiers = [ # For OS-independent nodes (works on all operating systems)