from dalle2_pytorch.dalle2_pytorch import DALLE2