Views
No views yet
1import torch
2from diffusers.models import SD3ControlNetModel
3
4
5def get_dummy_components_controlnet():
6 torch.manual_seed(0)
7 controlnet = SD3ControlNetModel(
8 sample_size=32,
9 patch_size=1,
10 in_channels=8,
11 num_layers=1,
12 attention_head_dim=8,
13 num_attention_heads=4,
14 joint_attention_dim=32,
15 caption_projection_dim=32,
16 pooled_projection_dim=64,
17 out_channels=8,
18 )
19 return controlnet
20
21
22if __name__ == "__main__":
23 controlnet = get_dummy_components_controlnet()
24 controlnet.push_to_hub("DavyMorgan/tiny-controlnet-sd3")