From 57a81bc0dbfbd730c02ee7bf393da3aabd0ac057 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sat, 28 Jun 2025 14:48:36 -0600 Subject: [PATCH] Update base model version for kontext meta --- .../diffusion_models/flux_kontext/flux_kontext.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py b/extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py index 8d4247c9..439c9a22 100644 --- a/extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py +++ b/extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py @@ -404,4 +404,7 @@ class FluxKontextModel(BaseModel): control_latent = self.encode_images(control_tensor).to(latents.device, latents.dtype) latents = torch.cat((latents, control_latent), dim=1) - return latents.detach() \ No newline at end of file + return latents.detach() + + def get_base_model_version(self): + return "flux.1_kontext" \ No newline at end of file