ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-29 10:41:28 +00:00
Code Issues 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

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

@@ -404,4 +404,7 @@ class FluxKontextModel(BaseModel):
control_latent = self.encode_images(control_tensor).to(latents.device, latents.dtype) control_latent = self.encode_images(control_tensor).to(latents.device, latents.dtype)
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"