.

Dockerfile

@@ -37,7 +37,7 @@ RUN python3 -m pip install torch_geometric
 #RUN python3 -m pip install torch-scatter -f https://data.pyg.org/whl/torch-2.5.1.html
 #RUN python3 -m pip install torch-sparse -f https://data.pyg.org/whl/torch-2.5.1.html
 #RUN python3 -m pip install torch-cluster -f https://data.pyg.org/whl/torch-2.5.1.html
-RUN python3 -m pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv xformers -f https://data.pyg.org/whl/torch-2.5.0+cpu.html
+RUN python3 -m pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv xformers==0.0.29.post1 -f https://data.pyg.org/whl/torch-2.5.0+cpu.html
 
 RUN python3 -m pip install pytorch-lightning yacs torchmetrics
 RUN python3 -m pip install performer-pytorch