Rename FluxControlNetModel -> DiffusersControlNetFlux

This commit is contained in:
Ryan Dick
2024-10-04 15:07:29 +00:00
committed by Kent Keirsey
parent 30f6034f88
commit bc0ded0a23

View File

@@ -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