github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Update base model version for kontext meta

Browse Source
This commit is contained in:
Jaret Burkett
2025-06-28 14:48:36 -06:00
parent 843be31138
commit 57a81bc0db
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py
View File

@@ -405,3 +405,6 @@ class FluxKontextModel(BaseModel):
latents = torch.cat((latents, control_latent), dim=1) latents = torch.cat((latents, control_latent), dim=1)
return latents.detach() return latents.detach()
def get_base_model_version(self):
return "flux.1_kontext"