diff --git a/dpkd/transformers/examples/flax/vision/requirements.txt b/dpkd/transformers/examples/flax/vision/requirements.txt
index 539ffdc6..52ee7ed2 100644
--- a/dpkd/transformers/examples/flax/vision/requirements.txt
+++ b/dpkd/transformers/examples/flax/vision/requirements.txt
@@ -3,6 +3,6 @@ jaxlib>=0.1.59
 flax>=0.3.5
 optax>=0.0.8
 -f https://download.pytorch.org/whl/torch_stable.html
-torch==1.11.0+cpu
+torch==2.9.1
 -f https://download.pytorch.org/whl/torch_stable.html
 torchvision==0.12.0+cpu