src/dalle_mini/model/__init__.py

from .configuration import DalleBartConfig
from .modeling import DalleBart
from .partitions import set_partitions
from .tokenizer import DalleBartTokenizer