colab AlphaFold蛋白质建模第五步无法运行

前面四步都很正常然后第五步运行了十几秒就开始报错

第四步出现tree（）opt/conda/ lib/python3.7/site-packages/haiku/_src/data_structures.py:37: FutureWarning: jax.tree_structure is deprecated, and will be removed in a future release. Use jax.tree_util.tree_structure instead.
PyTreeDef = type(jax.tree_structure(None))

小白求教，请问问题有没有解决呢？谢谢