DavyMorgan/tiny-controlnet-sd35

Pipeline generated with

import torch
from diffusers.models import SD3ControlNetModel


def get_dummy_components_controlnet():
    torch.manual_seed(0)
    controlnet = SD3ControlNetModel(
        sample_size=32,
        patch_size=1,
        in_channels=8,
        num_layers=1,
        attention_head_dim=8,
        num_attention_heads=4,
        joint_attention_dim=32,
        caption_projection_dim=32,
        pooled_projection_dim=64,
        out_channels=8,
        qk_norm="rms_norm",
        dual_attention_layers=(0,),
    )
    return controlnet


if __name__ == "__main__":
    controlnet = get_dummy_components_controlnet()
    controlnet.push_to_hub("DavyMorgan/tiny-controlnet-sd35")