File size: 649 Bytes
8abce2e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3243703
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
Pipeline generated with 

```python
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,
    )
    return controlnet


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