61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
from .registry import register_backend
|
|
from ..pro_edit_utils import build_fallback_latent_payload, build_native_conditioning_payload, set_conditioning_values
|
|
|
|
|
|
class FluxProEditBackend:
|
|
backend_name = "flux"
|
|
|
|
def supports(self, profile):
|
|
return str(profile.get("family") or "").strip().upper().startswith("FLUX")
|
|
|
|
def prepare(self, *, vae, work_image, work_mask, positive, negative, workspace, routing, **_kwargs):
|
|
if routing.execution_path != "native_edit":
|
|
latent = build_fallback_latent_payload(
|
|
vae,
|
|
work_image,
|
|
work_mask,
|
|
latent_source=f"pro_edit_fallback_{workspace['edit_scope']}",
|
|
)
|
|
return {
|
|
"latent": latent,
|
|
"positive": positive,
|
|
"negative": negative,
|
|
"backend_hints": {
|
|
"backend_name": self.backend_name,
|
|
"routing_reason": routing.routing_reason,
|
|
},
|
|
}
|
|
|
|
latent, concat_latent_image, concat_mask = build_native_conditioning_payload(
|
|
vae,
|
|
work_image,
|
|
work_mask,
|
|
latent_source=f"pro_edit_prepare_{workspace['edit_scope']}",
|
|
masked_fill_value=0.0,
|
|
)
|
|
values = {
|
|
"concat_latent_image": concat_latent_image,
|
|
"concat_mask": concat_mask,
|
|
"edit_backend": self.backend_name,
|
|
}
|
|
return {
|
|
"latent": latent,
|
|
"positive": set_conditioning_values(positive, values),
|
|
"negative": set_conditioning_values(negative, values),
|
|
"backend_hints": {
|
|
"backend_name": self.backend_name,
|
|
"routing_reason": routing.routing_reason,
|
|
},
|
|
}
|
|
|
|
def prepare_bridge(self, *, positive, negative, flux_guidance=None, **kwargs):
|
|
return {
|
|
"positive": positive,
|
|
"negative": negative,
|
|
"flux_guidance": flux_guidance,
|
|
**kwargs,
|
|
}
|
|
|
|
|
|
register_backend(FluxProEditBackend())
|