Update base model version for kontext meta

2025-06-28 14:48:36 -06:00 · 2025-06-28 14:48:36 -06:00 · 57a81bc0db
parent 843be31138
commit 57a81bc0db
1 changed files with 4 additions and 1 deletions
--- 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() 
+        return latents.detach() 
+
+    def get_base_model_version(self):
+        return "flux.1_kontext"