For everyone having a dependency issue on colab hosted env in cell 1.2 and later:

replace the line at the end of cell 1.2

!python -m pip -q install xformers

with

!python -m pip -q install xformers==0.0.21

save the notebook, delete and restart the runtime