boris's picture
feat(train): no batch dimension with pjit
df1fe19
raw
history blame
156 Bytes
from .configuration import DalleBartConfig
from .modeling import DalleBart
from .partitions import set_partitions
from .tokenizer import DalleBartTokenizer