Skip to content
This repository was archived by the owner on Feb 7, 2025. It is now read-only.

Commit ad973b8

Browse files
WarvitoWarvito
authored
Add ControlNet (#358)
* Add additional connections for controlnets Signed-off-by: Walter Hugo Lopez Pinaya <ianonimato@hotmail.com> * [WIP] Add ControlNet Signed-off-by: Walter Hugo Lopez Pinaya <ianonimato@hotmail.com> * [WIP] Add ControlNet Signed-off-by: Walter Hugo Lopez Pinaya <ianonimato@hotmail.com> * [WIP] Add ControlNet Signed-off-by: Walter Hugo Lopez Pinaya <ianonimato@hotmail.com> * [WIP] Add ControlNet Signed-off-by: Walter Hugo Lopez Pinaya <ianonimato@hotmail.com> * Add ControlNet Signed-off-by: Walter Hugo Lopez Pinaya <ianonimato@hotmail.com> --------- Signed-off-by: Walter Hugo Lopez Pinaya <ianonimato@hotmail.com>
1 parent d4ab8e8 commit ad973b8

4 files changed

Lines changed: 490 additions & 0 deletions

File tree

‎generative/networks/nets/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
from __future__ import annotations
1313

1414
from .autoencoderkl import AutoencoderKL
15+
from .controlnet import ControlNet
1516
from .diffusion_model_unet import DiffusionModelUNet
1617
from .patchgan_discriminator import MultiScalePatchDiscriminator, PatchDiscriminator
1718
from .transformer import DecoderOnlyTransformer

0 commit comments

Comments
 (0)