Update requirements

app.py

@@ -35,8 +35,8 @@ def load_predictor(version: str = "v1.8.1: Best Sharpness"):
     if version not in predictors:
         yoso_version = MODEL_VERSIONS[version]
         print(f"Loading StableNormal with {yoso_version}...")
-        predictor = torch.hub.load("Stable-X/StableNormal", "StableNormal_turbo", 
-                                   trust_repo=True, yoso_version=yoso_version)
+        predictor = torch.hub.load("./StableNormal", "StableNormal_turbo", 
+                                   trust_repo=True, yoso_version=yoso_version, source='local')
         predictors[version] = predictor
         print(f"Successfully loaded {version}")
     return predictors[version]

requirements.txt

@@ -18,7 +18,7 @@ cycler==0.12.1
 dataclasses-json==0.6.6
 datasets==2.19.1
 Deprecated==1.2.14
-diffusers==0.28.0
+diffusers==0.35.0
 dill==0.3.8
 dnspython==2.6.1
 email_validator==2.1.1
@@ -37,7 +37,7 @@ h11==0.14.0
 httpcore==1.0.5
 httptools==0.6.1
 httpx==0.27.0
-huggingface-hub==0.23.0
+huggingface-hub==0.36.0
 idna==3.7
 imageio==2.34.1
 imageio-ffmpeg==0.5.0
@@ -63,12 +63,12 @@ nvidia-cublas-cu12==12.1.3.1
 nvidia-cuda-cupti-cu12==12.1.105
 nvidia-cuda-nvrtc-cu12==12.1.105
 nvidia-cuda-runtime-cu12==12.1.105
-nvidia-cudnn-cu12==8.9.2.26
+nvidia-cudnn-cu12==9.1.0.70
 nvidia-cufft-cu12==11.0.2.54
 nvidia-curand-cu12==10.3.2.106
 nvidia-cusolver-cu12==11.4.5.107
 nvidia-cusparse-cu12==12.1.0.106
-nvidia-nccl-cu12==2.19.3
+nvidia-nccl-cu12==2.20.5
 nvidia-nvjitlink-cu12==12.5.40
 nvidia-nvtx-cu12==12.1.105
 orjson==3.10.3
@@ -103,18 +103,18 @@ semantic-version==2.10.0
 shellingham==1.5.4
 six==1.16.0
 sniffio==1.3.1
-spaces==0.28.3
+spaces
 starlette==0.37.2
 sympy==1.12.1
-tokenizers==0.15.2
+tokenizers==0.20.3
 tomlkit==0.12.0
 toolz==0.12.1
-torch==2.2.0
-torchvision==0.17.0
+torch==2.4.0
+torchvision==0.19.0
 tqdm==4.66.4
-transformers==4.36.1
+transformers==4.45.0
 trimesh==4.0.5
-triton==2.2.0
+triton==3.0.0
 typer==0.12.3
 typing-inspect==0.9.0
 typing_extensions==4.11.0
@@ -126,7 +126,7 @@ uvloop==0.19.0
 watchfiles==0.22.0
 websockets==11.0.3
 wrapt==1.16.0
-xformers==0.0.24
+xformers
 xxhash==3.4.1
 yarl==1.9.4
 zipp==3.19.1