Rename FluxControlNetModel -> DiffusersControlNetFlux

This commit is contained in:
Ryan Dick
2024-10-04 15:07:29 +00:00
parent 459cf52043
commit 3578010ba4

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