m4/training/types.py

from enum import Enum


class DatasetNames(Enum):
    PMD = "pmd"
    LAION = "laion"
    CM4 = "cm4"
    WIKI = "wiki"


class DatasetTypes(Enum):
    WEB_DOCUMENTS = "wd"
    IMAGE_CAPTION_PAIRS = "icp"