skip xformers if exists
This commit is contained in:
+14
@@ -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
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user