ntrolNet model. This model inherits from FlaxModelMixin. Check the superclass documentation for it’s generic methods
implemented for all models (such as downloading or saving). This model is also a Flax Linen flax.linen.Module
subclass. Use it as a regular Flax Linen module and refer to the Flax documentation for all matters related to its
general usage and behavior. Inherent JAX features such as the following are supported: Just-In-Time (JIT) 