back to torch 2.4

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,3 +1,7 @@
+--extra-index-url https://download.pytorch.org/whl/cu124
+torch==2.4.0
+torchaudio==2.4.0
+torchvision==0.19.0
 aiofiles==24.1.0
 aiohappyeyeballs==2.6.1
 aiohttp==3.11.16
@@ -17,7 +21,8 @@ charset-normalizer==3.4.1
 click==8.1.8
 coloredlogs==15.0.1
 comfy-cli==1.3.8
-comfyui_frontend_package==1.15.13
+comfyui_frontend_package==1.17.11
+comfyui_workflow_templates==0.1.3
 contourpy==1.3.2
 cookiecutter==2.6.0
 cryptography==44.0.2
@@ -75,20 +80,20 @@ mpmath==1.3.0
 multidict==6.4.3
 networkx==3.4.2
 numba==0.61.2
-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
+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
 nvidia-cudnn-cu12==9.1.0.70
-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-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-cusparselt-cu12==0.6.2
-nvidia-nccl-cu12==2.21.5
-nvidia-nvjitlink-cu12==12.4.127
-nvidia-nvtx-cu12==12.4.127
+nvidia-nccl-cu12==2.20.5
+nvidia-nvjitlink-cu12==12.4.99
+nvidia-nvtx-cu12==12.4.99
 onnx==1.17.0
 onnxruntime==1.21.0
 onnxruntime-gpu==1.21.0
@@ -104,7 +109,7 @@ prettytable==3.16.0
 prompt_toolkit==3.0.51
 propcache==0.3.1
 protobuf==6.30.2
-psutil==7.0.0
+psutil==5.9.8
 pycparser==2.22
 pydantic==2.11.3
 pydantic_core==2.33.1
@@ -140,6 +145,7 @@ 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
@@ -151,14 +157,10 @@ 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.2.0
+triton==3.0.0
 typer==0.15.2
 types-python-dateutil==2.9.0.20241206
 typing-inspection==0.4.0
@@ -173,4 +175,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