reverting

face_enhance.py

@@ -3,7 +3,7 @@ import random
 import sys
 from typing import Sequence, Mapping, Any, Union
 import torch
-import spaces
+# import spaces
 COMFYUI_PATH = "./ComfyUI"
 
 """
@@ -313,7 +313,7 @@ def save_comfy_images(images, output_dirs):
         pil_image = Image.fromarray(numpy_image)
         pil_image.save(output_dirs[idx])
 
-@spaces.GPU
+# @spaces.GPU
 def face_enhance(face_image: str, input_image: str, output_image: str, dist_image: str = None, positive_prompt: str = "", id_weight: float = 0.75):
     initialize_models()  # Ensure models are loaded
     main(face_image, input_image, output_image, dist_image, positive_prompt, id_weight)

requirements.txt

@@ -1,7 +1,3 @@
---extra-index-url https://download.pytorch.org/whl/cu121
-torch==2.1.2
-torchaudio==2.1.2
-torchvision==0.16.2
 aiofiles==24.1.0
 aiohappyeyeballs==2.6.1
 aiohttp==3.11.16
@@ -21,8 +17,7 @@ charset-normalizer==3.4.1
 click==8.1.8
 coloredlogs==15.0.1
 comfy-cli==1.3.8
-comfyui_frontend_package==1.17.11
-comfyui_workflow_templates==0.1.3
+comfyui_frontend_package==1.15.13
 contourpy==1.3.2
 cookiecutter==2.6.0
 cryptography==44.0.2
@@ -80,20 +75,20 @@ mpmath==1.3.0
 multidict==6.4.3
 networkx==3.4.2
 numba==0.61.2
-numpy==1.26.4
-nvidia-cublas-cu12==12.4.2.65
-nvidia-cuda-cupti-cu12==12.4.99
-nvidia-cuda-nvrtc-cu12==12.4.99
-nvidia-cuda-runtime-cu12==12.4.99
+numpy==2.2.4
+nvidia-cublas-cu12==12.4.5.8
+nvidia-cuda-cupti-cu12==12.4.127
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cuda-runtime-cu12==12.4.127
 nvidia-cudnn-cu12==9.1.0.70
-nvidia-cufft-cu12==11.2.0.44
-nvidia-curand-cu12==10.3.5.119
-nvidia-cusolver-cu12==11.6.0.99
-nvidia-cusparse-cu12==12.3.0.142
+nvidia-cufft-cu12==11.2.1.3
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cusparse-cu12==12.3.1.170
 nvidia-cusparselt-cu12==0.6.2
-nvidia-nccl-cu12==2.20.5
-nvidia-nvjitlink-cu12==12.4.99
-nvidia-nvtx-cu12==12.4.99
+nvidia-nccl-cu12==2.21.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.4.127
 onnx==1.17.0
 onnxruntime==1.21.0
 onnxruntime-gpu==1.21.0
@@ -109,7 +104,7 @@ prettytable==3.16.0
 prompt_toolkit==3.0.51
 propcache==0.3.1
 protobuf==6.30.2
-psutil==5.9.8
+psutil==7.0.0
 pycparser==2.22
 pydantic==2.11.3
 pydantic_core==2.33.1
@@ -145,7 +140,6 @@ six==1.17.0
 smmap==5.0.2
 sniffio==1.3.1
 soundfile==0.13.1
-spaces==0.35.0
 spandrel==0.4.1
 starlette==0.46.2
 stringzilla==3.12.4
@@ -157,10 +151,14 @@ timm==1.0.15
 tokenizers==0.21.1
 toml==0.10.2
 tomlkit==0.13.2
+torch==2.6.0
+torchaudio==2.6.0
+torchsde==0.2.6
+torchvision==0.21.0
 tqdm==4.67.1
 trampoline==0.1.2
 transformers==4.51.3
-triton==3.0.0
+triton==3.2.0
 typer==0.15.2
 types-python-dateutil==2.9.0.20241206
 typing-inspection==0.4.0
@@ -175,4 +173,4 @@ wcwidth==0.2.13
 websocket-client==1.8.0
 websockets==15.0.1
 wrapt==1.17.2
-yarl==1.20.0
+yarl==1.20.0