run prediction error

[gmestrallet]

---

ModuleNotFoundError Traceback (most recent call last)
in <cell line: 7>()
5 from colabfold.download import download_alphafold_params, default_data_dir
6 from colabfold.utils import setup_logging
----> 7 from colabfold.batch import get_queries, run
8
9 from pathlib import Path

8 frames
/usr/local/lib/python3.10/dist-packages/jax/_src/lib/init.py in
74 import jaxlib.xla_client as xla_client
75 import jaxlib.lapack as lapack
---> 76 import jaxlib.pocketfft as pocketfft
77
78 xla_extension = xla_client._xla

ModuleNotFoundError: No module named 'jaxlib.pocketfft'

---

NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.

To view examples of installing some common dependencies, click the
"Open Examples" button below.