.

Dockerfile

@@ -31,14 +31,13 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip
 RUN python3 -m pip install numba==0.58.1
 # packages without conda
 # RUN python3 -m pip install --no-cache-dir torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
-RUN python3 -m pip install torch==2.5.1 torchvision torchaudio
+RUN python3 -m pip install torch==2.5.0 torchvision torchaudio
 RUN python3 -m pip install torch_geometric
 #RUN python3 -m pip install  torch_scatter torch_sparse
 #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 -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 -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
@@ -63,7 +62,6 @@ RUN python3 -m pip install xxhash
 RUN python3 -m pip install tables
 RUN python3 -m pip install tensorboard
 RUN python3 -m pip install plotly
-RUN python3 -m pip install xformers
 RUN python3 -m pip install fastjet
 RUN python3 -m pip install gradio
 RUN python3 -m pip install huggingface_hub