ModuleNotFoundError: No module named 'diff_dalle'

[hamdjalil]

tried training on a colab with the following command:

!python /content/Diff-DALLE/scripts/train_generator.py --data_dir /content/drive/MyDrive/path/to/dataset $MODEL_FLAGS $DIFFUSION_FLAGS $TRAIN_FLAGS

which gives the following error:

Traceback (most recent call last):
File "/content/Diff-DALLE/scripts/train_generator.py", line 7, in
from diff_dalle import dist_util, logger
ModuleNotFoundError: No module named 'diff_dalle'