|
|
--- |
|
|
library_name: diffusers |
|
|
--- |
|
|
|
|
|
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, |
|
|
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") |
|
|
``` |
