Add ControlNet

generative/networks/nets/__init__.py

@@ -12,6 +12,7 @@
 from __future__ import annotations
 
 from .autoencoderkl import AutoencoderKL
+from .controlnet import ControlNet
 from .diffusion_model_unet import DiffusionModelUNet
 from .patchgan_discriminator import MultiScalePatchDiscriminator, PatchDiscriminator
 from .transformer import DecoderOnlyTransformer