Update requirements.txt

requirements.txt

@@ -6,24 +6,26 @@ tqdm
 psutil
 einops
 torchsde
-torchaudio
 spandrel
 kornia>=0.7.1
 gguf
 replicate
-numba  # Added for was-node-suite-comfyui
+numba
 
-# PyTorch and AI ecosystem
-torch==2.4.0
-torchvision
+# PyTorch and AI ecosystem (installed via install.sh)
+# torch==2.4.0
+# torchvision
+# torchaudio
 accelerate
 diffusers
 transformers
 xformers
 fairscale>=0.4.4
 timm>=0.4.12
-pytorch-lightning  # Added for CatVTON
-inplace-abn==1.1.0  # Added for CatVTON
+pytorch-lightning
+
+# CUDA-dependent packages (installed after CUDA setup)
+inplace-abn==1.1.0
 
 # WAS Node Suite dependencies
 git+https://github.com/WASasquatch/img2texture.git
@@ -45,33 +47,7 @@ yacs>=0.1.8
 tabulate
 pycocotools>=2.0.2
 termcolor>=1.1
+git+https://github.com/facebookresearch/[email protected]  # Base detectron2
 git+https://github.com/facebookresearch/[email protected]#subdirectory=projects/DensePose
-git+https://github.com/facebookresearch/[email protected]  # Added base detectron2 for CatVTON
-
-# Image Processing
-opencv-python-headless[ffmpeg]
-scikit-image>=0.20.0
-scikit-learn
-matplotlib
-
-# Development & Build Tools
-Ninja==1.11.1.1
-setuptools
-cloudpickle
-tensorboard
-pydot
-cmake
-gitpython
-
-# Hugging Face Tools
-huggingface_hub
-
-# Comfyui-tensorops dependencies
-surrealist
-boto3==1.34.86
-redis
-fal-client
 
-# Additional utilities
-PyYAML
-blend_modes
+# Rest of your dependencies...