from autotrain.trainers.dreambooth.dreambooth_deprecated import train as train_dreambooth_deprecated
from autotrain.trainers.dreambooth.main import train


XL_MODELS = ["stabilityai/stable-diffusion-xl-base-0.9"]
