Source code for deepfloyd_if_xl_dreambooth_lora_dog

from mmengine.config import read_base

with read_base():
    from .._base_.datasets.pokemon_blip_if import *
    from .._base_.default_runtime import *
    from .._base_.models.deepfloyd_if_xl_lora import *
    from .._base_.schedules.stable_diffusion_50e import *


[docs]optim_wrapper = dict(
    optimizer=dict(type="AdamW", lr=5e-6, weight_decay=1e-2),
    clip_grad=dict(max_norm=1.0))