diff --git a/invokeai/backend/flux/controlnet/diffusers_controlnet_flux.py b/invokeai/backend/flux/controlnet/diffusers_controlnet_flux.py index d34950fe0b..0446362bc2 100644 --- a/invokeai/backend/flux/controlnet/diffusers_controlnet_flux.py +++ b/invokeai/backend/flux/controlnet/diffusers_controlnet_flux.py @@ -23,7 +23,7 @@ class FluxControlNetOutput(BaseOutput): controlnet_single_block_samples: Tuple[torch.Tensor] -class FluxControlNetModel(ModelMixin, ConfigMixin, PeftAdapterMixin): +class DiffusersControlNetFlux(ModelMixin, ConfigMixin, PeftAdapterMixin): _supports_gradient_checkpointing = True @register_to_config