fixing rfdiffusion

Author: sokrypton

rf/examples/diffusion.ipynb

@@ -62,7 +62,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",
         "  os.system(\"wget -qnc https://files.ipd.uw.edu/krypton/ananas\")\n",
         "  os.system(\"chmod +x ananas\")\n",