[ComfyUI] ModuleNotFoundError: No module named 'comfy_aimdo' in Colab

[Meleksu8]

Hi Ben,

I encountered an error while trying to run the fast_stable_diffusion_ComfyUI.ipynb notebook on Google Colab today.

Despite doing a fresh install (deleting the ComfyUI folder and reinstalling), I keep getting the following error during the boot process:

Error Log:

Traceback (most recent call last):
File "/content/gdrive/MyDrive/ComfyUI/main.py", line 55, in
import cuda_malloc
File "/content/gdrive/MyDrive/ComfyUI/cuda_malloc.py", line 6, in
import comfy_aimdo.control
ModuleNotFoundError: No module named 'comfy_aimdo'

It seems like a missing dependency or a path issue in the latest update. Thanks for your hard work!

[modpotato]

same issue via swarmui

[Meleksu8]

same issue via swarmui

you can fix it with
!pip install comfy_aimdo

[alexjl90210]

I reverted back these files

in ComfyUI folder
execution.py
main.py

in ComfyUI/Comfy folder
model_management.py
model_patcher.py
ops.py

I grabbed come clean versions of them before the AIMDO lines were added. Got mine from the portable version of ComfyUI.
Once I used those files it started normally without the error. Ran in Colab again without issues.