No module named 'diffusers.models.unet_2d_condition'

[atharvaraut0]

1. Running comfyui using google colab.
2. The virtual environment is using python version 3.11.11

I have installed deforum using

%cd $WORKSPACE
%cd ./custom_nodes/
!git clone https://github.com/XmYx/deforum-comfy-nodes.git
!git pull
!pip install pydub
!pip install deforum
!pip install pydantic

Snippet of the 'error' :

File "/usr/local/lib/python3.11/dist-packages/deforum/utils/model_mixing.py", line 6, in <module>
    from diffusers.models.unet_2d_condition import UNet2DConditionModel
ModuleNotFoundError: No module named 'diffusers.models.unet_2d_condition'

Cannot import /content/drive/MyDrive/ComfyUI/custom_nodes/deforum-comfy-nodes module for custom nodes: No module named 'diffusers.models.unet_2d_condition'

Import times for custom nodes:
   0.0 seconds: /content/drive/MyDrive/ComfyUI/custom_nodes/websocket_image_save.py
   0.1 seconds: /content/drive/MyDrive/ComfyUI/custom_nodes/ComfyUI-Custom-Scripts
   0.8 seconds (IMPORT FAILED): /content/drive/MyDrive/ComfyUI/custom_nodes/deforum-comfy-nodes