diff --git a/README.md b/README.md
index 5e3553c..fc79fef 100644
--- a/README.md
+++ b/README.md
@@ -350,4 +350,4 @@ We will try our best to release
       primaryClass={cs.CV},
       url={https://arxiv.org/abs/2410.10629},
     }
-```
+```
\ No newline at end of file
diff --git a/pyproject.toml b/pyproject.toml
index 9eed756..2e1788e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,7 +25,7 @@ dependencies = [
     "gradio",
     "image-reward",
     "ipdb",
-    "mmcv==1.7.2",
+    "mmcv==2.2.0",
     "omegaconf",
     "opencv-python",
     "optimum",
@@ -38,13 +38,13 @@ dependencies = [
     "tensorboard",
     "tensorboardX",
     "timm",
-    "torchaudio==2.4.0",
-    "torchvision==0.19",
+    "torchaudio==2.5.0",
+    "torchvision==0.20.0",
     "transformers",
-    "triton==3.0.0",
+    "triton==3.1.0",
     "wandb",
     "webdataset",
-    "xformers==0.0.27.post2",
+    "xformers",
     "yapf",
     "spaces",
     "matplotlib",