fixing pytorch error

Author: sokrypton

rf/examples/diffusion_foldcond.ipynb

@@ -78,7 +78,7 @@
         "  os.system(\"pip install git+https://github.com/NVIDIA/dllogger#egg=dllogger\")\n",
         "  # 17Mar2024: adding --no-dependencies to avoid installing nvidia-cuda-* dependencies\n",
         "  os.system(\"pip install --no-dependencies dgl==2.0.0 -f https://data.dgl.ai/wheels/cu121/repo.html\")\n",
-        "  os.system(\"pip install --no-dependencies e3nn==0.3.3 opt_einsum_fx\")\n",
+        "  os.system(\"pip install --no-dependencies e3nn==0.5.5 opt_einsum_fx\")\n",
         "  os.system(\"cd RFdiffusion/env/SE3Transformer; pip install .\")\n",
         "\n",
         "  # extras\n",