Colab Notebook setup not working

[Zeclown]

Running the colab Notebook setup Simplified_Disco_Diffusion.ipynb yields the following error

ModuleNotFoundError Traceback (most recent call last)
in ()
107 # Import DD helper modules
108 sys.path.append(PROJECT_DIR)
--> 109 import dd, dd_args
110
111 # Unsure about these:

/content/gdrive/MyDrive/disco-diffusion-1/dd.py in ()
37 from deepdiff import DeepHash
38 import sqlite3
---> 39 from torchmetrics import RetrievalFallOut
40 from tqdm.notebook import tqdm
41 from twilio.rest import Client

ModuleNotFoundError: No module named 'torchmetrics'

---

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.

[dave20012]

!pip install torchmetrics

manually inserted in to the enviroment setup
on the colab notebook

fixed the error for me