diff --git a/pyproject.toml b/pyproject.toml
index 3e70a0c..024c70f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,13 +1,19 @@
 [project]
 dependencies = [
+    "scikit-image~=0.25",
+    "spconv-cu120==2.3.6",
+    "torch==2.5.1",
+    "torchaudio==2.5.1",
+    "torchsde==0.2.6",
+    "torchvision==0.20.1",
+    "trimesh~=4.7",
+    "xformers~=0.0",
     "diffusers~=0.28",
     "accelerate~=1.9",
-    "triton~=3.2",
+    "triton=~3.1",
     "kornia~=0.8",
     "timm~=0.6",
-    "transformers~=4.46",
-    "trimesh~=4.7",
-    "scikit-image~=0.25",
+    "transformers~=4.46"
 ]
 
 
diff --git a/requirements.txt b/requirements.txt
index d5dfe80..7800de0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,20 +1,19 @@
 --extra-index-url https://download.pytorch.org/whl/cu124
+scikit-image~=0.25
+spconv-cu120==2.3.6
+torch==2.5.1
+torchaudio==2.5.1
+torchsde==0.2.6
+torchvision==0.20.1
+trimesh~=4.7
+xformers~=0.0
+
 # For StableNormal
 diffusers~=0.28
 accelerate~=1.9
-triton
+triton~=3.1
 
 # For BirefNet
 kornia~=0.8
 timm~=0.6
-transformers~=4.46
-
-trimesh~=4.7
-scikit-image~=0.25
-xformers~=0.0
-
-torch==2.5.1
-torchaudio==2.5.1
-torchvision==0.20.1
-torchsde
-spconv-cu120==2.3.6
\ No newline at end of file
+transformers~=4.46
\ No newline at end of file