ModuleNotFoundError: No module named 'models'

I've been getting this error when trying out the model:

---------------------------------------------------------------------------
ModuleNotFoundError Traceback (most recent call last)
[](https://localhost:8080/#) in ()
1 import data
2 import torch
----> 3 from models import imagebind_model
4 from models.imagebind_model import ModalityType
5

ModuleNotFoundError: No module named 'models'

---------------------------------------------------------------------------
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.