diff --git a/modules/rfdiffusion/notebooks/requirements.txt b/modules/rfdiffusion/notebooks/requirements.txt index abee44cf..35296040 100644 --- a/modules/rfdiffusion/notebooks/requirements.txt +++ b/modules/rfdiffusion/notebooks/requirements.txt @@ -7,7 +7,7 @@ e3nn==0.3.3 wandb==0.12.0 pynvml==11.0.0 decorator==5.1.0 -torch==1.11.0+cu113 +torch==2.7.0 torchvision==0.12.0+cu113 torchaudio==0.11.0 mlflow==2.15.1