DavyMorgan commited on
Commit
8abce2e
·
verified ·
1 Parent(s): 0cc0be0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -3
README.md CHANGED
@@ -1,3 +1,28 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Pipeline generated with
2
+
3
+ ```python
4
+ import torch
5
+ from diffusers.models import SD3ControlNetModel
6
+
7
+
8
+ def get_dummy_components_controlnet():
9
+ torch.manual_seed(0)
10
+ controlnet = SD3ControlNetModel(
11
+ sample_size=32,
12
+ patch_size=1,
13
+ in_channels=8,
14
+ num_layers=1,
15
+ attention_head_dim=8,
16
+ num_attention_heads=4,
17
+ joint_attention_dim=32,
18
+ caption_projection_dim=32,
19
+ pooled_projection_dim=64,
20
+ out_channels=8,
21
+ )
22
+ return controlnet
23
+
24
+
25
+ if __name__ == "__main__":
26
+
27
+ controlnet = get_dummy_components_controlnet()
28
+ controlnet.push_to_hub("DavyMorgan/tiny-controlnet-sd3")