diffengine.engine.hooks.t2i_adapter_save_hook

Module Contents

Classes

T2IAdapterSaveHook

T2I Adapter Save Hook.

class diffengine.engine.hooks.t2i_adapter_save_hook.T2IAdapterSaveHook

Bases: mmengine.hooks.Hook

T2I Adapter Save Hook.

Save T2I-Adapter weights with diffusers format and pick up weights from checkpoint.

priority = VERY_LOW

before_save_checkpoint(runner, checkpoint)

Before save checkpoint hook.

Args:

runner (Runner): The runner of the training, validation or testing

process.

checkpoint (dict): Model’s checkpoint.

Parameters:

• runner (mmengine.runner.Runner) –

• checkpoint (dict) –

Return type:

None