skip xformers if exists

This commit is contained in:
VjumpKunG
2025-12-24 12:25:39 +07:00
parent efdd7c57c5
commit 91a58167cc
2 changed files with 15 additions and 1 deletions
+14
View File
@@ -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()
+1 -1
View File
@@ -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)