ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-01 03:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #343 from davertor/fix_kontext_bs

Browse Source 
fix: Guidance incorrect shape
This commit is contained in:
Jaret Burkett
2025-07-20 12:00:55 -06:00
committed by GitHub
parent 8610c6ed7f a77ba5a089
commit b1bff66d52
1 changed files with 2 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

@@ -291,7 +291,8 @@ class FluxKontextModel(BaseModel):
else: else:
guidance = torch.tensor( guidance = torch.tensor(
[guidance_embedding_scale], device=self.device_torch) [guidance_embedding_scale], device=self.device_torch)
guidance = guidance.expand(latent_model_input.shape[0]) # Expand guidance to match original batch_size
guidance = guidance.expand(bs)
else: else:
guidance = None guidance = None